def save_training(self, X_train, y_train, X_test, y_test):
utils.mkdir('model')
self.save_model(X_train, 'model/X_train.pkl')
self.save_model(X_test, 'model/X_test.pkl')
self.save_model(y_train, 'model/y_train.pkl')
self.save_model(y_test, 'model/y_test.pkl')
评论列表
文章目录