class TestDevRemoteBSTClassifier(AsyncTestCase):
"""Test the RemoteBSTClassifier against the dev server"""
def setUp(self):
super().setUp()
conf = get_conf()
self.model_params = {
"hiddenLayers": [5, 6, 7, 8],
"learningRate": 0.4,
}
url = conf['aimetrics']['dev']['hosts']['ai']['base_url']
self.clf = RemoteBSTClassifier(url, "bnn",
model_params=self.model_params)
self.assertIsNotNone(self.clf)
@gen_test
def test_async_fit(self):
yield self.clf.async_fit(train_X, train_y)
# test params
m = yield self.clf.get_model()
self.assertEqual(2+ len(self.model_params['hiddenLayers']),
len(m["model"]["layers"]))
def tearDown(self):
if self.clf.model_id:
self.clf.destroy_model()
