pycaret学習したモデルで予想を行う

いよいよ機械学習の目的である予想を行っていきます。pycaretの予想を行うためのテストデータを準備します。pycaretでテストデータを準備していなかったので、今回はtitanicのテストデータを別のサイトから持ってきました。

テストデータの読み込み

import numpy as np
import pandas as pd

test_data = pd.read_csv("test.csv") # どっかから持ってきたtitaniceのテストデータ

test_data.head()

テストデータ

予想を行う

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()

Labelに予想結果が出たの図

最後に

pycaret使えば数行のコードで簡単に予想が行えた。という話でございました。次は各関数のパラメータを細かく調べていこうかな。