Skip to content
Last updated

Gluon Predict

This notebook is used for prediction using a prediction model built by gluon_train notebook.

Sample Output

If the same column name to label that is used in training (e.g., y) exists in input_table , prediction result is resulted in predicted_y with predicted prefix_. If the label column does not exist in input_table , prediction is resulted in y column.

The columns predicted_proba and predicted_probabilities are results for binary or multiclass classifications. For binary classification, predict_proba is probability of positive class. You can get the probability of a negative class by 1.0 - predict_proba. For multiclass classification, predict_proba is probability of predicted class.

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}

If rowid_column is provided, it only appends prediction results to the rowid column and the resulting size of table will be small. It is assumed that user_id or product_id is unique for the rowid_column.

Workflow Example

For a sample workflow, see 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

Parameters

Parameter nameParameter on ConsoleDescriptionDefault Value
docker.task_memDocker Task MemTask memory size. Available values are 64g, 128g (default), 256g, 384g, or 512g depending on your contracted tiers128g
model_nameModel NamePrediction model name-
input_tableInput TableTD table name used for test data-
output_tableOutput TableTD table name to export the prediction result-
output_modeOutput ModeOutput mode for exporting output_table: overwrite or replace or append. Usually, there is no need to specify and append for semi-realtime prediction with rowid_filteroverwrite
ignore_columnsIgnore ColumnsColumns to ignore when building a prediction modeltime
rowid_columnRowid ColumnThe rowid (primary key) column. Note that when rowid_column is specified, only rowid column and prediction result columns are displayed in the output table.None
rowid_filterRowid FilterFiltering rows for prediction by rowid column valuesNone
export_leaderboardExport LeaderboardExport leaderboard as a TD table, if specifiedNone
export_feature_importanceExport Feature ImportanceExport feature importance as a TD table, if specifiedNone
hide_table_contentsHide Table ContentsSuppress showing table contentsfalse
shared_modelShared ModelSpecify a shared model UUIDNone
audience_nameAudience NameAudience parent (master) segment name to merge an attribute tableNone
foreign_keyForeign KeyForeign key in an Audience parent (master) table to merge an attribute tableNone