iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update baselines

Browse Source
This commit is contained in:
D-X-Y
2021-05-27 23:54:14 +08:00
parent 33ea7ca87a
commit 585ded31da
5 changed files with 36 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_synthetic_env.py
View File

@@ -5,7 +5,7 @@
#####################################################
import unittest
from xautodl.datasets.math_core import ConstantFunc, ComposedSinFunc
from xautodl.datasets.math_core import ConstantFunc, ComposedSinSFunc
from xautodl.datasets.synthetic_core import SyntheticDEnv
@@ -13,7 +13,7 @@ class TestSynethicEnv(unittest.TestCase):
"""Test the synethtic environment."""
def test_simple(self):
mean_generator = ComposedSinFunc(constant=0.1)
mean_generator = ConstantFunc(constant=0.1)
std_generator = ConstantFunc(constant=0.5)
dataset = SyntheticDEnv([mean_generator], [[std_generator]], num_per_task=5000)
print(dataset)
@@ -21,7 +21,7 @@ class TestSynethicEnv(unittest.TestCase):
self.assertEqual(tau.shape, (5000, 1))
def test_length(self):
mean_generator = ComposedSinFunc(constant=0.1)
mean_generator = ComposedSinSFunc({0: 1, 1: 1, 2: 3})
std_generator = ConstantFunc(constant=0.5)
dataset = SyntheticDEnv([mean_generator], [[std_generator]], num_per_task=5000)
self.assertEqual(len(dataset), 100)