![見出し画像](https://assets.st-note.com/production/uploads/images/77809736/rectangle_large_type_2_e7b9023bff3ddcff2b264b700558731c.png?width=800)
Photo by
sama_box
pycaret学習したモデルで予想を行う
いよいよ機械学習の目的である予想を行っていきます。pycaretの予想を行うためのテストデータを準備します。pycaretでテストデータを準備していなかったので、今回はtitanicのテストデータを別のサイトから持ってきました。
テストデータの読み込み
import numpy as np
import pandas as pd
test_data = pd.read_csv("test.csv") # どっかから持ってきたtitaniceのテストデータ
test_data.head()
![](https://assets.st-note.com/img/1651661466034-ULL7HXq4NY.png?width=800)
予想を行う
pycaretのpredict_modelを使って予想を行います。Labelに予想結果が出ているのが分かります。
from pycaret.classification import predict_model
test_pred_ridge = predict_model(final_ridge, data=test_data) # predict_modelで予想を行う
test_pred_ridge.head()
![](https://assets.st-note.com/img/1651661626481-SWCPpGTgH2.png?width=800)
最後に
pycaret使えば数行のコードで簡単に予想が行えた。という話でございました。次は各関数のパラメータを細かく調べていこうかな。
この記事が気に入ったらサポートをしてみませんか?