【问题标题】:How to use scikit-learn linear regression without using split?如何在不使用拆分的情况下使用 scikit-learn 线性回归?
【发布时间】:2021-07-26 10:33:04
【问题描述】:

我有 2 个名为 train.csv 和 test.csv 的 CSV。

两个文件的结构相同,我想用 train.csv 作为训练数据,用 test.csv 作为测试数据。

问题是,我在任何地方都找不到如何在不使用拆分的情况下使用 scikit-learn 线性回归,我找到的每个教程/文档都使用函数 train_test_split(),但如果我理解正确,它用于拆分一个文件(假设 data.csv)作为训练和测试数据。

有可能吗?如果不是,我可以使用什么替代方案?

【问题讨论】:

  • 只是不要使用拆分功能。将train.csv 读取为训练数据集,将test.csv 读取为测试数据集。

标签: python csv scikit-learn


【解决方案1】:

如果您有单独的traintest 数据,

定义X_trainy_train

  • X_train 是不包括目标变量的特征
# Sudo Code
X_train = train.drop(target, axis=1)
  • y_train 是目标变量
# Sudo Code
y_train = train[target]
from sklearn.linear_model import LinearRegression
reg = LinearRegression().fit(X_train, y_train)

【讨论】:

    猜你喜欢
    • 2018-04-22
    • 2018-07-31
    • 2017-12-31
    • 2016-05-16
    • 2017-11-29
    • 2016-11-14
    • 2019-05-14
    • 2017-12-25
    • 2019-03-08
    相关资源
    最近更新 更多