Skip to content
Last updated

Gluon Predict

このノートブックは、gluon_train ノートブックで構築された予測モデルを使用して予測を行うために使用されます。

サンプル出力

トレーニングで使用されたラベルと同じカラム名(例: y)が input_table に存在する場合、予測結果は predicted というプレフィックスを付けた predicted_y として出力されます。ラベルカラムが input_table に存在しない場合、予測結果は y カラムに出力されます。

predicted_proba および predicted_probabilities カラムは、二値分類または多クラス分類の結果です。二値分類の場合、predict_proba はポジティブクラスの確率です。ネガティブクラスの確率は 1.0 - predict_proba で取得できます。多クラス分類の場合、predict_proba は予測されたクラスの確率です。

ypredicted_ypredicted_probapredicted_probabilities
yesno0.29304519295692444{"no": 0.706954836845398, "yes": 0.29304519295692444}
nono0.00487339636310935{"no": 0.9951266050338745, "yes": 0.00487339636310935}
............
yesyes0.5266873240470886{"no": 0.4733126759529114, "yes": 0.5266873240470886}

rowid_column が指定されている場合、rowid カラムに予測結果のみを追加し、結果のテーブルサイズは小さくなります。rowid_column には、user_idproduct_id のような一意の値を想定しています。

ワークフロー例

サンプルワークフローについては、Treasure Boxesを参照してください。

+gluon_predict:
  ml_predict>:
    notebook: gluon_predict
    model_name: gluon_model
    input_table: ml_datasets.bank_marketing_test
    output_table: ml_test.predicted

パラメータ

パラメータ名コンソール上のパラメータ説明デフォルト値
docker.task_memDocker Task Memタスクメモリサイズ。契約されているティアに応じて、64g、128g(デフォルト)、256g、384g、または512gの値を使用できます128g
model_nameModel Name予測モデル名-
input_tableInput Tableテストデータとして使用されるTDテーブル名-
output_tableOutput Table予測結果をエクスポートするTDテーブル名-
output_modeOutput Modeoutput_table をエクスポートする際の出力モード: overwrite、replace、または append。通常、指定する必要はなく、rowid_filter を使用したセミリアルタイム予測ではappendを使用しますoverwrite
ignore_columnsIgnore Columns予測モデル構築時に無視するカラムtime
rowid_columnRowid Columnrowid(主キー)カラム。rowid_column が指定されている場合、出力テーブルにはrowidカラムと予測結果カラムのみが表示されることに注意してください。None
rowid_filterRowid Filterrowid カラム値による予測対象の行のフィルタリングNone
export_leaderboardExport Leaderboard指定されている場合、リーダーボードをTDテーブルとしてエクスポートしますNone
export_feature_importanceExport Feature Importance指定されている場合、特徴量の重要度をTDテーブルとしてエクスポートしますNone
hide_table_contentsHide Table Contentsテーブルの内容の表示を抑制しますfalse
shared_modelShared Model共有モデルのUUIDを指定しますNone
audience_nameAudience Name属性テーブルをマージするためのAudienceペアレント(マスター)セグメント名None
foreign_keyForeign Key属性テーブルをマージするためのAudienceペアレント(マスター)テーブル内の外部キーNone