{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "$id": "https://schema.rtemis.org/decomposition/umap/v1/schema.json",
  "title": "rtemis UMAPConfig",
  "description": "Uniform Manifold Approximation and Projection. See setup_UMAP.",
  "type": "object",
  "additionalProperties": false,
  "properties": {
    "k": {
      "type": "integer",
      "minimum": 1,
      "default": 2,
      "description": "Number of components to extract."
    },
    "n_neighbors": {
      "type": "integer",
      "minimum": 1,
      "default": 15,
      "description": "Number of neighbors."
    },
    "init": {
      "type": "string",
      "default": "spectral",
      "description": "Initialization type (see `uwot::umap` `init`)."
    },
    "metric": {
      "type": "string",
      "enum": ["euclidean", "cosine", "manhattan", "hamming", "categorical"],
      "default": "euclidean",
      "description": "Distance metric."
    },
    "n_epochs": {
      "type": ["integer", "null"],
      "minimum": 1,
      "default": null,
      "description": "Number of epochs. NULL = algorithm default."
    },
    "learning_rate": {
      "type": "number",
      "minimum": 0,
      "default": 1,
      "description": "Learning rate."
    },
    "scale": {
      "type": "boolean",
      "default": true,
      "description": "Scale input data before UMAP."
    }
  }
}
