TensorFlow.NET/test/TensorFlowNET.UnitTest/Basics/RandomTest.cs at master · LiveFly/TensorFlow.NET · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Tensorflow.NumPy;
using System;
using System.Linq;
using Tensorflow;
using static Tensorflow.Binding;

namespace TensorFlowNET.UnitTest.Basics
{
    [TestClass]
    public class RandomTest
    {
        /// <summary>
        /// Test the function of setting random seed
        /// This will help regenerate the same result
        /// </summary>
        [TestMethod]
        public void TFRandomSeedTest()
        {
            var initValue = np.arange(6).reshape((3, 2));
            tf.set_random_seed(1234);
            var a1 = tf.random_uniform(1);
            var b1 = tf.random_shuffle(tf.constant(initValue));

            // This part we consider to be a refresh
            tf.set_random_seed(10);
            tf.random_uniform(1);
            tf.random_shuffle(tf.constant(initValue));

            tf.set_random_seed(1234);
            var a2 = tf.random_uniform(1);
            var b2 = tf.random_shuffle(tf.constant(initValue));
            Assert.AreEqual(a1.numpy(), a2.numpy());
            Assert.AreEqual(b1.numpy(), b2.numpy());
        }

        /// <summary>
        /// compare to Test above, seed is also added in params
        /// </summary>
        [TestMethod, Ignore]
        public void TFRandomSeedTest2()
        {
            var initValue = np.arange(6).reshape((3, 2));
            tf.set_random_seed(1234);
            var a1 = tf.random_uniform(1, seed:1234);
            var b1 = tf.random_shuffle(tf.constant(initValue), seed: 1234);

            // This part we consider to be a refresh
            tf.set_random_seed(10);
            tf.random_uniform(1);
            tf.random_shuffle(tf.constant(initValue));

            tf.set_random_seed(1234);
            var a2 = tf.random_uniform(1);
            var b2 = tf.random_shuffle(tf.constant(initValue));
            Assert.AreEqual(a1, a2);
            Assert.AreEqual(b1, b2);
        }

        /// <summary>
        /// This part we use funcs in tf.random rather than only tf
        /// </summary>
        [TestMethod]
        public void TFRandomRaodomSeedTest()
        {
            tf.set_random_seed(1234);
            var a1 = tf.random.normal(1);
            var b1 = tf.random.truncated_normal(1);

            // This part we consider to be a refresh
            tf.set_random_seed(10);
            tf.random.normal(1);
            tf.random.truncated_normal(1);

            tf.set_random_seed(1234);
            var a2 = tf.random.normal(1);
            var b2 = tf.random.truncated_normal(1);

            Assert.AreEqual(a1.numpy(), a2.numpy());
            Assert.AreEqual(b1.numpy(), b2.numpy());
        }

        /// <summary>
        /// compare to Test above, seed is also added in params
        /// </summary>
        [TestMethod, Ignore]
        public void TFRandomRaodomSeedTest2()
        {
            tf.set_random_seed(1234);
            var a1 = tf.random.normal(1, seed:1234);
            var b1 = tf.random.truncated_normal(1);

            // This part we consider to be a refresh
            tf.set_random_seed(10);
            tf.random.normal(1);
            tf.random.truncated_normal(1);

            tf.set_random_seed(1234);
            var a2 = tf.random.normal(1, seed:1234);
            var b2 = tf.random.truncated_normal(1, seed:1234);

            Assert.AreEqual(a1, a2);
            Assert.AreEqual(b1, b2);
        }
    }
}