12.18 每日总结(今天机器学习大作业)
最近在写机器学习大作业。混凝土配方预测其成品的承重强度。
学习时长3小时
下面是测试集测试的代码
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score
# 定义特征和标签
X = data.iloc[:, :-1].values
y = data['Compressive_Strength_Grade'].values
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 训练模型
model = LogisticRegression()
model.fit(X_train, y_train)
# 预测
y_pred = model.predict(X_test)

浙公网安备 33010602011771号