[Python 100日チャレンジ] Day 92 - scikit-learnの準備

scikit-learnとは

scikit-learnは、Pythonで機械学習を行うための代表的なライブラリです。分類、回帰、クラスタリング、前処理、評価などの機能がそろっています。

pip install scikit-learn

インポートする

from sklearn.datasets import load_iris

iris = load_iris()
print(iris.data.shape)
print(iris.target_names)

学習の基本形

scikit-learnでは、多くのモデルが同じ流れで使えます。

model.fit(X_train, y_train)
predictions = model.predict(X_test)

fit()で学習し、predict()で予測します。

今日のゴール

scikit-learnの使い方は、データ準備、モデル作成、fit、predictが基本です。以降の機械学習実践に向けて環境を整えましょう。