wehub-resource-sync
18 lines
580 B
Plaintext
18 lines
580 B
Plaintext
name: cuML RF test
|
|
|
|
conda_env: envs/conda.yaml
|
|
|
|
entry_points:
|
|
hyperopt:
|
|
parameters:
|
|
fpath: {type: str}
|
|
algo: {type: str, default: 'tpe'}
|
|
command: "python src/rf_test/train.py --fpath={fpath} --algo={algo}"
|
|
simple:
|
|
parameters:
|
|
fpath: {type: str}
|
|
n_estimators: {type: int}
|
|
max_features: {type: float}
|
|
max_depth: {type: int}
|
|
command: "python src/rf_test/train_simple.py --fpath={fpath} --n_estimators={n_estimators} --max_features={max_features} --max_depth={max_depth}"
|