Spaces:
Running
Running
| { | |
| "models": [ | |
| { | |
| "id": "logistic_regression", | |
| "label": "Logistic regression", | |
| "estimator": "LogisticRegression", | |
| "fixed": { | |
| "max_iter": 1000 | |
| }, | |
| "params": [ | |
| { | |
| "name": "C", | |
| "label": "Regularisation", | |
| "kind": "number", | |
| "min": 0.01, | |
| "max": 100, | |
| "default": 1.0 | |
| } | |
| ] | |
| }, | |
| { | |
| "id": "knn", | |
| "label": "k-nearest neighbours", | |
| "estimator": "KNeighborsClassifier", | |
| "fixed": {}, | |
| "params": [ | |
| { | |
| "name": "n_neighbors", | |
| "label": "Neighbours", | |
| "kind": "integer", | |
| "min": 1, | |
| "max": 30, | |
| "default": 5 | |
| } | |
| ] | |
| }, | |
| { | |
| "id": "svm", | |
| "label": "Support vector machine", | |
| "estimator": "SVC", | |
| "fixed": {}, | |
| "params": [ | |
| { | |
| "name": "kernel", | |
| "label": "Kernel", | |
| "kind": "choice", | |
| "options": [ | |
| { | |
| "value": "linear", | |
| "label": "Linear" | |
| }, | |
| { | |
| "value": "rbf", | |
| "label": "RBF" | |
| }, | |
| { | |
| "value": "poly", | |
| "label": "Polynomial" | |
| } | |
| ], | |
| "default": "rbf" | |
| }, | |
| { | |
| "name": "C", | |
| "label": "Regularisation", | |
| "kind": "number", | |
| "min": 0.01, | |
| "max": 100, | |
| "default": 1.0 | |
| }, | |
| { | |
| "name": "gamma", | |
| "label": "Kernel width", | |
| "kind": "number", | |
| "min": 0.01, | |
| "max": 100, | |
| "default": 1.0, | |
| "when": { | |
| "kernel": [ | |
| "rbf", | |
| "poly" | |
| ] | |
| } | |
| } | |
| ] | |
| }, | |
| { | |
| "id": "decision_tree", | |
| "label": "Decision tree", | |
| "estimator": "DecisionTreeClassifier", | |
| "fixed": { | |
| "random_state": 0 | |
| }, | |
| "params": [ | |
| { | |
| "name": "max_depth", | |
| "label": "Maximum depth", | |
| "kind": "integer", | |
| "min": 1, | |
| "max": 15, | |
| "nullable": "Unlimited", | |
| "default": null | |
| } | |
| ] | |
| }, | |
| { | |
| "id": "random_forest", | |
| "label": "Random forest", | |
| "estimator": "RandomForestClassifier", | |
| "fixed": { | |
| "random_state": 0 | |
| }, | |
| "params": [ | |
| { | |
| "name": "n_estimators", | |
| "label": "Trees", | |
| "kind": "integer", | |
| "min": 1, | |
| "max": 200, | |
| "default": 100 | |
| }, | |
| { | |
| "name": "max_depth", | |
| "label": "Maximum depth", | |
| "kind": "integer", | |
| "min": 1, | |
| "max": 15, | |
| "nullable": "Unlimited", | |
| "default": null | |
| } | |
| ] | |
| }, | |
| { | |
| "id": "gradient_boosting", | |
| "label": "Gradient boosting", | |
| "estimator": "GradientBoostingClassifier", | |
| "fixed": { | |
| "random_state": 0 | |
| }, | |
| "params": [ | |
| { | |
| "name": "n_estimators", | |
| "label": "Stages", | |
| "kind": "integer", | |
| "min": 1, | |
| "max": 200, | |
| "default": 100 | |
| }, | |
| { | |
| "name": "learning_rate", | |
| "label": "Learning rate", | |
| "kind": "number", | |
| "min": 0.01, | |
| "max": 1, | |
| "default": 0.1 | |
| } | |
| ] | |
| } | |
| ] | |
| } | |