utils.functions.MLModelLib

Module Contents

Functions

build_xgb_ml_model(h2o_train_df, h2o_validation_df, ...)

build_auto_ml_model(h2o_train_df[, target_col, ...])

get_h2o_datatype_map([rcEntityPropertiesMap, ds_name])

predict_model_h2o(h2o_model, prediction_df_h2o, target_col)

get_prediction_stats(df_prediction, target_col[, ...])

get_precision_recall(model_perf[, label_map])

gen_ml_report(h2o_ml_model, h2o_test_df, target_col)

convert_lc_pdf_2_h2o_df(lc_input_df)

utils.functions.MLModelLib.build_xgb_ml_model(h2o_train_df, h2o_validation_df, target_col, ignore_cols, model_id='unnamed_model')
utils.functions.MLModelLib.build_auto_ml_model(h2o_train_df, target_col='', ignore_cols=[], project_name='automl', include_algos=['GBM', 'XGBoost'], nfolds=4, max_models=5, seed=1234, h2o_leaderb_df=None)
utils.functions.MLModelLib.get_h2o_datatype_map(rcEntityPropertiesMap={}, ds_name=None)
utils.functions.MLModelLib.predict_model_h2o(h2o_model, prediction_df_h2o, target_col)
utils.functions.MLModelLib.get_prediction_stats(df_prediction, target_col, prediction_col='predict')
utils.functions.MLModelLib.get_precision_recall(model_perf, label_map={})
utils.functions.MLModelLib.gen_ml_report(h2o_ml_model, h2o_test_df, target_col)
utils.functions.MLModelLib.convert_lc_pdf_2_h2o_df(lc_input_df)