14.python 机器学习入门(scikit-learn)

scikit-learn 是 Python 中最常用的机器学习库之一，它提供了大量的机器学习算法和工具，可以帮助我们进行数据预处理、特征选择、模型训练和评估等任务。

以下是使用 scikit-learn 进行机器学习的简单步骤：

1. 准备数据

首先需要准备好训练集和测试集。训练集用于训练模型，测试集则用于评估模型的性能。在 scikit-learn 中，可以使用 train_test_split() 函数将数据集划分为训练集和测试集。

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

# 加载 iris 数据集
iris = load_iris()

# 将数据集划分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(
    iris.data, iris.target, test_size=0.3, random_state=42)

2. 特征工程

接下来需要对数据进行预处理和特征选择。在 scikit-learn 中，可以使用各种数据预处理和特征选择方法，如标准化、归一化、降维等。

例如，我们可以使用 StandardScaler 对数据进行标准化处理：

from sklearn.preprocessing import StandardScaler

# 创建 StandardScaler 对象
scaler = StandardScaler()

# 对训练集和测试集进行标准化处理
X_train_scaled = scaler.fit_transform(X_train)
X_test_scaled = scaler.transform(X_test)

3. 模型训练

使用 scikit-learn 中提供的各种机器学习算法进行模型训练。以支持向量机（SVM）为例：

from sklearn.svm import SVC

# 创建 SVM 对象
svm = SVC(kernel='linear', C=1, random_state=42)

# 使用训练集训练模型
svm.fit(X_train_scaled, y_train)

4. 模型评估

使用测试集对训练好的模型进行性能评估。在 scikit-learn 中，可以使用各种评估指标，如准确率、精度、召回率等。

例如，我们可以使用 accuracy_score 函数计算模型的准确率：

from sklearn.metrics import accuracy_score

# 对测试集进行预测
y_pred = svm.predict(X_test_scaled)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)

以上就是使用 scikit-learn 进行机器学习的基本流程。需要注意的是，不同的机器学习算法和任务可能需要采用不同的数据预处理方法、特征选择方法和评估指标，具体要根据实际情况选择。