Commit ·
d446464
1
Parent(s): daf21c8
fix: more specific web search query for overfitting suggestions
Browse files- 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 |
-
|
|
|
|
| 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]:
|