rapids-higgs-excerpt-2.py

parser = argparse.ArgumentParser()

# Hyper-parameters
parser.add_argument('--n_estimators',        type=int,   default=20)
parser.add_argument('--max_depth',           type=int,   default=16)
parser.add_argument('--n_bins',              type=int,   default=8)
parser.add_argument('--split_criterion',     type=int,   default=0)
parser.add_argument('--split_algo',          type=int,   default=0)
parser.add_argument('--bootstrap',           type=bool,  default=True)
parser.add_argument('--bootstrap_features',  type=bool,  default=False)
parser.add_argument('--max_leaves',          type=int,   default=-1)
parser.add_argument('--max_features',        type=float, default=0.2)

# SageMaker parameters
parser.add_argument('--model_dir',        type=str)
parser.add_argument('--model_output_dir', type=str,   default='/opt/ml/output/')
parser.add_argument('--data_dir',         type=str,   default='/opt/ml/input/data/dataset/')

args = parser.parse_args()
main(args)