用python写出利用回归模型预测鲍鱼的年龄的代码
时间: 2023-12-30 22:03:45 浏览: 81
机器学习实战 鲍鱼年龄预测 knn svm 逻辑回归 有代码可运行
5星 · 资源好评率100%
好的,以下是一个简单的例子:
```python
import pandas as pd
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
# 导入数据
abalone = pd.read_csv('abalone.csv')
# 数据预处理
abalone['Sex'] = abalone['Sex'].map({'M': 1, 'F': 0, 'I': 0.5})
X = abalone.drop('Rings', axis=1)
y = abalone['Rings']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建模型并训练
reg = LinearRegression()
reg.fit(X_train, y_train)
# 预测年龄
new_data = pd.DataFrame({'Sex': [1], 'Length': [0.55], 'Diameter': [0.43], 'Height': [0.13], 'Whole weight': [0.51], 'Shucked weight': [0.22], 'Viscera weight': [0.10], 'Shell weight': [0.15]})
predicted_age = reg.predict(new_data)
print(predicted_age)
```
在这个例子中,我们使用了鲍鱼数据集,并使用线性回归模型来预测鲍鱼的年龄。我们首先将数据集导入,然后进行了一些数据预处理,例如将性别转换为数值型变量。接着,我们将数据集分成训练集和测试集,并使用训练集来训练模型。最后,我们用新的数据来预测鲍鱼的年龄,并将结果打印出来。
阅读全文