scikit-learn常用库与函数整理

需要提前导入的库：

numpy

import numpy as np

绘图

import matplotlib.pyplot as plt

数据集

from sklearn import datasets

各种机器学习库：

kNN算法

from sklearn.neighbors import KNeighborsClassifier

线性回归

from sklearn.linear_model import LinearRegression

PCA

from sklearn.decomposition import PCA
pca = PCA(n_components=2)
pca.fit(X_train)
X_train_reduction = pca.transform(X_train)
X_test_reduction = pca.transform(X_test)

逻辑回归

from sklearn.linear_model import LogisticRegression

SVM

from sklearn.svm import LinearSVC