docs: use generic predictor terminology in training output
Browse files
router.py
CHANGED
|
@@ -137,7 +137,7 @@ class R2Router:
|
|
| 137 |
with open(os.path.join(path, "training_data", "labels.json")) as f:
|
| 138 |
labels = json.load(f)
|
| 139 |
|
| 140 |
-
print(f"Training
|
| 141 |
|
| 142 |
quality_knns = {}
|
| 143 |
token_knns = {}
|
|
@@ -173,7 +173,7 @@ class R2Router:
|
|
| 173 |
token_knns[model_name] = tknn
|
| 174 |
n_token += 1
|
| 175 |
|
| 176 |
-
print(f"Trained {n_quality} quality
|
| 177 |
|
| 178 |
model_prices = {
|
| 179 |
mn: cfg["output_price_per_million"]
|
|
|
|
| 137 |
with open(os.path.join(path, "training_data", "labels.json")) as f:
|
| 138 |
labels = json.load(f)
|
| 139 |
|
| 140 |
+
print(f"Training router on {len(X_train)} samples (k={k})...")
|
| 141 |
|
| 142 |
quality_knns = {}
|
| 143 |
token_knns = {}
|
|
|
|
| 173 |
token_knns[model_name] = tknn
|
| 174 |
n_token += 1
|
| 175 |
|
| 176 |
+
print(f"Trained {n_quality} quality predictors + {n_token} token predictors for {len(quality_knns)} models.")
|
| 177 |
|
| 178 |
model_prices = {
|
| 179 |
mn: cfg["output_price_per_million"]
|