multiclass.yaml

- node_type: embedding
  target_metric: retrieval_hit_rate
  search_space:
    - module_name: retrieval
      k: [10]
      embedder_name:
        - sentence-transformers/all-MiniLM-L6-v2
        - avsolatorio/GIST-small-Embedding-v0
- node_type: scoring
  target_metric: scoring_roc_auc
  search_space:
    - module_name: knn
      k: [5, 10]
      weights: [uniform, distance, closest]
    - module_name: linear
    - module_name: dnnc
      cross_encoder_name:
        - cross-encoder/ms-marco-MiniLM-L-6-v2
        - avsolatorio/GIST-small-Embedding-v0
      k: [1, 3]
      train_head: [false, true]
    - module_name: sklearn
      embedder_name:
        - sergeyzh/rubert-tiny-turbo
      clf_name:
        - LogisticRegression
        - RandomForestClassifier
    - module_name: rerank
      k: [ 5, 10 ]
      weights: [uniform, distance, closest]
      m: [ 2, 3 ]
      cross_encoder_name:
        - cross-encoder/ms-marco-MiniLM-L-6-v2
- node_type: decision
  target_metric: decision_accuracy
  search_space:
    - module_name: threshold
      thresh: [0.5, [0.5, 0.5, 0.5, 0.5]]
    - module_name: tunable
    - module_name: argmax
    - module_name: jinoos