(input_data)
| 13 | |
| 14 | @staticmethod |
| 15 | def predict(input_data): |
| 16 | assert RegressionPredictor.__model, \ |
| 17 | "Use 'RegressionPredictor()' to initialize before using 'RegressionPredictor.predict()'" |
| 18 | |
| 19 | x = input_data.reshape(1, 28, 28) |
| 20 | return RegressionPredictor.__model.predict(x).flatten().tolist() |
| 21 | |
| 22 | |
| 23 | class CNNPredictor(object): |