如何在机器学习模型中设置最小值?

2024-06-28 14:34:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个模型可以根据学生的表现来预测他们的成绩。
但该模型预测了一些负值的等级。
因此,如何将最小值设置为零,如果预测等级为负,则返回0

X_train,X_test , y_train , y_test =train_test_split(X,y,test_size=0.2)
model= LinearRegression()
model.fit(X_train,y_train)
predictions = model.predict(X_test)


Tags: 模型testsizemodeltrainpredict学生fit