def testTrain(self): model = 'tests/test_data/minimodel.json' data = 'tests/test_data/hdf5/alphago-vs-lee-sedol-features.hdf5' output = 'tests/test_data/.tmp.training/' args = [model, data, output, '--epochs', '1'] run_training(args) os.remove(os.path.join(output, 'metadata.json')) os.remove(os.path.join(output, 'shuffle.npz')) os.remove(os.path.join(output, 'weights.00000.hdf5')) os.rmdir(output)
def testTrain(self): model = 'tests/test_data/minimodel.json' data = 'tests/test_data/hdf5/alphago-vs-lee-sedol-features.hdf5' output = 'tests/test_data/.tmp.training/' args = [model, data, output, '--epochs', '1'] run_training(args) os.remove(os.path.join(output, 'metadata.json')) os.remove(os.path.join(output, 'shuffle.npz')) os.remove(os.path.join(output, 'weights.00.hdf5')) os.rmdir(output)
def run_supervised_policy_training(): run_training(*arguments)