对一个csv数据集进行训练的代码
时间: 2024-02-15 18:04:08 浏览: 134
训练数据集:Advertising.csv
好的,以下是一个基本的使用Python和Pandas库对CSV数据集进行训练的代码:
```python
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
# 读取CSV文件
data = pd.read_csv('your_dataset.csv')
# 将数据集分为训练集和测试集
X = data.drop('target_column_name', axis=1)
y = data['target_column_name']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 训练决策树模型
model = DecisionTreeClassifier()
model.fit(X_train, y_train)
# 在测试集上测试模型
accuracy = model.score(X_test, y_test)
print(f"Accuracy: {accuracy}")
```
这段代码假设你的CSV数据集文件名为`your_dataset.csv`,其中包含一个需要预测的目标列`target_column_name`和其他特征列。它使用Pandas库读取数据集,并使用`train_test_split`函数将数据集分为训练集和测试集。然后,它使用Scikit-learn库中的`DecisionTreeClassifier`训练了一个决策树模型,并在测试集上测试了模型的准确性。
阅读全文