simplify docs configs (#113)

Samoed · web-flow · commit 2c5f7f047d48 · 2025-02-01T09:47:44.000+03:00
diff --git a/autointent/_datafiles/default-multiclass-config.yaml b/autointent/_datafiles/default-multiclass-config.yaml
@@ -6,7 +6,7 @@
       k: [10]
       embedder_name:
         - avsolatorio/GIST-small-Embedding-v0
-        - infgrad/stella-base-en-v2
+        - sergeyzh/rubert-tiny-turbo
 - node_type: scoring
   target_metric: scoring_roc_auc
   search_space:
@@ -16,7 +16,6 @@
     - module_name: linear
     - module_name: dnnc
       cross_encoder_name:
-        - BAAI/bge-reranker-base
         - cross-encoder/ms-marco-MiniLM-L-6-v2
       k: [1, 3, 5, 10]
 - node_type: decision
diff --git a/autointent/_datafiles/inference-config-example.yaml b/autointent/_datafiles/inference-config-example.yaml
@@ -2,7 +2,7 @@
   module_name: retrieval
   module_config:
     k: 10
-    model_name: infgrad/stella-base-en-v2
+    model_name: sergeyzh/rubert-tiny-turbo
   load_path: .
 - node_type: scoring
   module_name: knn
diff --git a/user_guides/advanced/02_search_space.py b/user_guides/advanced/02_search_space.py
@@ -24,7 +24,7 @@
 knn_module = {
     "module_name": "knn",
     "k": [1, 5, 10, 50],
-    "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],
+    "embedder_name": ["sergeyzh/rubert-tiny-turbo"],
 }
 
 # %% [markdown]
@@ -78,7 +78,7 @@
             {
                 "module_name": "retrieval",
                 "k": [10],
-                "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],
+                "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],
             }
         ],
     },
@@ -90,7 +90,7 @@
             {"module_name": "linear"},
             {
                 "module_name": "dnnc",
-                "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"],
+                "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],
                 "k": [1, 3, 5, 10],
             },
         ],
diff --git a/user_guides/advanced/04_reporting.py b/user_guides/advanced/04_reporting.py
@@ -14,7 +14,7 @@
             {
                 "module_name": "retrieval",
                 "k": [10],
-                "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],
+                "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],
             }
         ],
     },
@@ -26,7 +26,7 @@
             {"module_name": "linear"},
             {
                 "module_name": "dnnc",
-                "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"],
+                "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],
                 "k": [1, 3, 5, 10],
             },
         ],

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`knn_module = {`
`25`	`25`	`"module_name": "knn",`
`26`	`26`	`"k": [1, 5, 10, 50],`
`27`		`- "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],`
	`27`	`+ "embedder_name": ["sergeyzh/rubert-tiny-turbo"],`
`28`	`28`	`}`
`29`	`29`
`30`	`30`	`# %% [markdown]`
`@@ -78,7 +78,7 @@`
`78`	`78`	`{`
`79`	`79`	`"module_name": "retrieval",`
`80`	`80`	`"k": [10],`
`81`		`- "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],`
	`81`	`+ "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],`
`82`	`82`	`}`
`83`	`83`	`],`
`84`	`84`	`},`
`@@ -90,7 +90,7 @@`
`90`	`90`	`{"module_name": "linear"},`
`91`	`91`	`{`
`92`	`92`	`"module_name": "dnnc",`
`93`		`- "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"],`
	`93`	`+ "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],`
`94`	`94`	`"k": [1, 3, 5, 10],`
`95`	`95`	`},`
`96`	`96`	`],`
Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@`
`14`	`14`	`{`
`15`	`15`	`"module_name": "retrieval",`
`16`	`16`	`"k": [10],`
`17`		`- "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"],`
	`17`	`+ "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],`
`18`	`18`	`}`
`19`	`19`	`],`
`20`	`20`	`},`
`@@ -26,7 +26,7 @@`
`26`	`26`	`{"module_name": "linear"},`
`27`	`27`	`{`
`28`	`28`	`"module_name": "dnnc",`
`29`		`- "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"],`
	`29`	`+ "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],`
`30`	`30`	`"k": [1, 3, 5, 10],`
`31`	`31`	`},`
`32`	`32`	`],`