【发布时间】:2020-10-27 13:29:43
【问题描述】:
我正在尝试根据学生的年龄来预测他们的班级,到目前为止一切都很好,但我上面显示的错误继续出现。
由于我是 python 机器学习的新手,这是我的第一个线性回归模型,我更喜欢不涉及类和对象的简单解决方案。
这是我的代码
import numpy as np
import pandas as pd
import pylab as pl
from sklearn import linear_model
import matplotlib.pyplot as plt
file = pd.read_csv('age and class cs version.csv')
age = file[['Age ']]
classes = file[['Class']]
age_dataframe = pd.DataFrame(age)
classes_dataframe = pd.DataFrame(classes)
regr = linear_model.LinearRegression()
model = regr.fit(age_dataframe,classes_dataframe)
new_age = 13
print(model.predict(new_age))
这是来自 csv 文件的数据
Age Class
7 1
11 5
14 9
14 8
15 10
16 12
17 12
15 8
9 3
【问题讨论】:
-
print(model.predict([[new_age]])) -
请发布完整的错误跟踪(不在此处,请更新您的帖子)。
-
您可能想要检查输入的形状。它们是 (some_number, ) 还是 (number, number)。
标签: python arrays pandas machine-learning