【发布时间】:2022-08-06 23:23:29
【问题描述】:
bach_sal = bach[\'Salary\']
masters_sal = masters[\'Salary\']
phd_sal = phd[\'Salary\']
deg_category_sal = [assoc_sal, bach_sal, masters_sal, phd_sal]
x = np.array(assoc[\'Person\'])
y = np.array(assoc_sal)
from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(x,y,test_size=0.7)
from sklearn.linear_model import LinearRegression
lr = LinearRegression()
lr.fit(x_train.reshape(-1,1),y_train)
print (lr.coef_)
print (lr.intercept_)
我收到了系数和 y 截距,但每次运行代码时,它们都会更改为不同的数字。请帮忙。谢谢你。
-
也许您可以尝试为您的代码/在哪里执行 train_test_split 设置一个随机种子。该方法随机选择您的数据的 %,因此它会在运行之间发生变化\\
标签: python machine-learning scikit-learn statistics data-science