def test_cross_validate(): # only tests if the code runs dataset = Dataset(kind='hplc', num_folds=5) model = BayesNeuralNet(scope='hplc', max_epochs=2) emulator = Emulator(dataset=dataset, model=model) emulator.cross_validate() emulator.save('emulator_test') shutil.rmtree('emulator_test')
def test_train_nn(): # only tests if the code dataset = Dataset(kind='hplc', num_folds=5) model = NeuralNet(scope='hplc', max_epochs=2) emulator = Emulator(dataset=dataset, model=model) emulator.train() emulator.save('emulator_test') shutil.rmtree('emulator_test')