(self)
| 40 | super(IrisTest, self).tearDown() |
| 41 | |
| 42 | def testTrainAndSaveNonSequential(self): |
| 43 | final_train_accuracy = iris.train(100, self._tmp_dir) |
| 44 | self.assertGreater(final_train_accuracy, 0.9) |
| 45 | |
| 46 | # Check that the model json file is created. |
| 47 | json.load(open(os.path.join(self._tmp_dir, 'model.json'), 'rt')) |
| 48 | |
| 49 | def testTrainAndSaveSequential(self): |
| 50 | final_train_accuracy = iris.train(100, self._tmp_dir, sequential=True) |