scikit-learn教程

scikit-learn（通常简称为sklearn）是Python中最受欢迎的机器学习库之一，它提供了各种监督和非监督学习算法的实现。下面是一个基本的教程，涵盖如何使用sklearn进行数据预处理、模型训练和评估。

1. 安装和导入包

首先确保安装了scikit-learn库，可以使用pip命令进行安装：

pip install -U scikit-learn

然后在Python脚本中导入所需的模块：

import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score, classification_report

2. 加载数据

sklearn自带了一些数据集，例如鸢尾花数据集（Iris dataset）：

from sklearn.datasets import load_iris
data = load_iris()
X = data.data
y = data.target

或者你可以加载自己的数据集：

df = pd.read_csv('your_dataset.csv')
X = df.drop('target_column', axis=1)
y = df['target_column']

3. 数据预处理

通常需要对数据进行预处理，包括分割数据集、填充缺失值、特征缩放等：

# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)# 特征缩放
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

4. 模型训练

选择一个模型，例如逻辑回归，并训练它：

model = LogisticRegression()
model.fit(X_train, y_train)

5. 模型预测

使用训练好的模型进行预测：

y_pred = model.predict(X_test)

6. 模型评估

评估模型的性能：

print("Accuracy:", accuracy_score(y_test, y_pred))
print(classification_report(y_test, y_pred))

7. 模型调优

可以使用交叉验证和网格搜索来调整模型参数：

from sklearn.model_selection import GridSearchCVparameters = {'C': [0.1, 1, 10]}
clf = GridSearchCV(LogisticRegression(), parameters)
clf.fit(X_train, y_train)# 最佳参数
print(clf.best_params_)

8. 模型保存与加载

使用joblib库保存和加载模型：

from sklearn.externals import joblib# 保存模型
joblib.dump(model, 'model.pkl')# 加载模型
model = joblib.load('model.pkl')

请注意，上面的joblib保存和加载代码示例适用于scikit-learn较旧版本。在scikit-learn 0.23及更高版本中，应使用joblib库直接进行保存和加载：

import joblib# 保存模型
joblib.dump(model, 'model.pkl')# 加载模型
model = joblib.load('model.pkl')

以上就是使用scikit-learn进行机器学习的基本流程。你可以根据具体需求替换模型、调整参数和执行更复杂的预处理步骤。