JiaqiXue commited on
Commit
3858b5d
·
verified ·
1 Parent(s): 9e08fe1

docs: use generic predictor terminology in training output

Browse files
Files changed (1) hide show
  1. router.py +2 -2
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 KNN (k={k}) on {len(X_train)} samples...")
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 KNNs + {n_token} token KNNs for {len(quality_knns)} models.")
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"]