我已经解决了多个问题,这些问题有助于将你的数据框分为训练和测试,使用 scikit,不使用等。
但我的问题是我有 2 个不同的 csv(来自不同年份的 2 个不同的数据框)。我想用一个作为火车,另一个作为测试?
如何对线性回归/任何模型执行此操作?
train
设置为 fit
模型。test
设置为 predict
训练后的输出。# Load the data
train = pd.read_csv('train.csv')
test = pd.read_csv('test.csv')
# Split features and value
# when trying to predict column "target"
X_train, y_train = train.drop("target"), train["target"]
X_test, y_test = test.drop("target"), test["target"]
# Fit (i.e. train) model
reg = LinearRegression()
reg.fit(X_train, y_train)
# Predict
pred = reg.predict(X_test)
# Score
accuracy = reg.score(X_test, y_test)
请问 Skillsmuggler X_train 和 X_Test 如何定义它,因为当我尝试这样做时,它说 NameError: name 'X_train' is not Defined
我无法编辑几乎就在那里的第一个答案。但缺少一些代码...
# Load the data
train = pd.read_csv('train.csv')
test = pd.read_csv('test.csv')
y_train = train[:, :1] #if y is only one column
X_train = train[:, 1:]
# 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)