rapids-higgs-excerpt-1.py

data = cudf.read_csv(data_dir+'HIGGS.csv', names=col_names, dtype=dtypes_ls)
X_train, X_test, y_train, y_test = train_test_split(data, 'label', train_size=0.70)
  
 hyperparams={
    'n_estimators'       : args.n_estimators,
    'max_depth'          : args.max_depth,
    'n_bins'             : args.n_bins,
    'split_criterion'    : args.split_criterion,
    'split_algo'         : args.split_algo,
    'bootstrap'          : args.bootstrap,
    'bootstrap_features' : args.bootstrap_features,
    'max_leaves'         : args.max_leaves,
    'max_features'       : args.max_features
    }
    
cu_rf = cuRF(**hyperparams)
cu_rf.fit(X_train, y_train)