Abdourakib commited on
Commit
d446464
·
1 Parent(s): daf21c8

fix: more specific web search query for overfitting suggestions

Browse files
Files changed (1) hide show
  1. agent/core.py +2 -1
agent/core.py CHANGED
@@ -537,7 +537,8 @@ class AutoMLAgent:
537
  warnings = self._train_result.get("overfitting_warnings", [])
538
  if warnings:
539
  best_name = self._train_result.get("best_name", "model")
540
- query = f"how to fix overfitting in {best_name} scikit-learn"
 
541
  try:
542
  search_results = web_search(query, num_results=5)
543
  if search_results and "error" not in search_results[0]:
 
537
  warnings = self._train_result.get("overfitting_warnings", [])
538
  if warnings:
539
  best_name = self._train_result.get("best_name", "model")
540
+ model_name = str(best_name).replace(" (tuned)", "").strip()
541
+ query = f"overfitting {model_name} regularization techniques machine learning"
542
  try:
543
  search_results = web_search(query, num_results=5)
544
  if search_results and "error" not in search_results[0]: