【发布时间】:2020-07-01 07:44:40
【问题描述】:
我在 Model 类中编写了这个简单随机森林回归的小型机器学习代码。在创建了这个类的一个对象之后,我打印了预测和准确度分数,同时我编写了一个代码来安排每 30 天的训练和每 7 天的测试。但我遇到了一个错误
代码:
import schedule
import time
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
import numpy as np
import pandas as pd
from main import data as df
class Model():
def __init__(self):
self.df = df
self.linear_reg = LinearRegression()
self.random_forest = RandomForestRegressor()
def split(self, test_size):
X = np.array(self.df[['age','experience','certificates']])
y = np.array(self.df['salary'])
self.X_train, self.X_test, self.y_train, self.y_test = train_test_split(X, y, test_size = test_size, random_state = 42)
def fit(self):
self.model = self.random_forest.fit(self.X_train, self.y_train)
def predict(self):
self.result = self.random_forest.predict(self.X_test)
print(self.result)
print("Accuracy: ", self.model.score(self.X_test, self.y_test))
if __name__ == '__main__':
model_instance = Model()
model_instance.split(0.2)
schedule.every(30).days.at("05:00").do(model_instance.fit())
schedule.every(7).days.at("05:00").do(model_instance.predict())
while 1:
schedule.run_pending()
time.sleep(1)
在这一行 schedule.every(30).days.at("05:00").do(model_instance.fit()) 我收到以下错误:the first argument must be callable
【问题讨论】:
标签: python class oop machine-learning scikit-learn