{
  "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
        }
      ]
    }
  ]
}