机器学习基础代码示例 🚀

安装依赖

pip install scikit-learn pandas numpy

数据加载示例

import pandas as pd

# 加载数据集
data = pd.read_csv('/data/iris.csv')  # 📁
features = data.iloc[:, :-1]
labels = data.iloc[:, -1]

print("特征样本：\n", features.head())
print("标签样本：\n", labels.head())

模型训练与预测

from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier

# 划分训练集与测试集
X_train, X_test, y_train, y_test = train_test_split(features, labels, test_size=0.2)

# 创建并训练模型
model = RandomForestClassifier()
model.fit(X_train, y_train)

# 预测
predictions = model.predict(X_test)
print("预测结果：", predictions)

扩展阅读

想要深入了解机器学习原理？点击这里查看机器学习入门教程 📚