scikit-learnのサンプルデータセットを使う

scikit-learnは機械学習ぼライブラリである。
また、scikit-learnにはあらかじめ学習されているサンプルデータセットがいくつか存在する。


以下、そのデータセットのうち手書き数字データを取得し図として描画した手順。

 

事前準備として以下のライブラリをインストールする

pip install scikit-learn

pip install matplotlib

scikit-learn・・・機械学習のライブラリ

matplotlib・・・画像を表示するライブラリ

 

 

参照設定のコード

#sklearnのデータセットを参照する
from sklearn import datasets
#pyplotを参照する
from matplotlib import pyplot

 

データセットを取得し、画像として描画するコード

#ライブラリで用意されている数字のデータセットを取得する
valux, valuy = datasets.load_digits(return_X_y=True)


#valuxの先頭(0番目)のデータを取得する
valux_aaa = valux[1]        #ここの数字を変えると別の画像が描画されます
 
#Figure(図全体)とAxes(座標軸)を同時に作成する
fighoge, axhoge = pyplot.subplots()

#取得したAxes(座標軸)を8ピクセル四方で描画する
axhoge.imshow(valux_aaa.reshape(8,8))

 


return_X_y=True・・・戻り値をタプル(組)にするかを指定する