我经历了多个问题,这些问题可帮助您将数据框分为使用scikit进行训练和测试,而无需其他操作。
但是我的问题是我有2个不同的csvs(来自不同年份的2个不同的数据帧)。我想将其中一个用作培训,将另一个用作测试?
如何对LinearRegression /任何模型执行此操作?
答案 0 :(得分:2)
train
设置为fit
的模型。test
设置为predict
的输出。# Load the data
train = pd.read_csv('train.csv')
test = pd.read_csv('test.csv')
# Fit (train) model
reg = LinearRegression()
reg.fit(X_train, y_train)
# Predict
pred = reg.predict(X_test)
# Score
accuracy = reg.socre(X_test, y_test)