使用Python建立线性回归模型并找到R2'd；无法让模型拟合或预测

Question

一些进口有几个原因

import pandas as pd
import numpy as np

我成功拆分了数据 -test(30%) 和 train(70%) 并将其分开：

X_train = df_train.drop(columns='Rating')
y_train = df_train.Rating
from sklearn.linear_model import LinearRegression
X_test = df_test.drop(columns='Rating')
y_test = df_test.Rating

到此为止一切都很好，那么

linreg = LinearRegression()
linreg.fit(X_train, y_train)

ValueError: could not convert string to float: 'GAME'

我肯定 Rating 列是浮动的

Answer 1

检查您的 df 第一行，它可能会在该位置再次重复标题。或者从第二排开始训练。