# AutoGluon Predictor constructor arguments # - see https://github.com/awslabs/autogluon/blob/master/tabular/src/autogluon/tabular/predictor/predictor.py#L51-L159 ag_predictor_args: eval_metric: roc_auc label: class # AutoGluon Predictor.fit arguments # - see https://github.com/awslabs/autogluon/blob/master/tabular/src/autogluon/tabular/predictor/predictor.py#L250-L590 ag_fit_args: hyperparameters: GBM: num_boost_round: 20 NN_TORCH: num_epochs: 2 num_bag_folds: 3 num_bag_sets: 1 num_stack_levels: 0 output_prediction_format: csv # predictions output format: csv or parquet feature_importance: true # calculate and save feature importance if true leaderboard: true # save leaderboard output if true