Spaces:
Sleeping
Sleeping
File size: 965 Bytes
a309487 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import time
class BenchmarkRunner:
def run(self, orchestrator, datasets):
results = []
for name, (df, target) in datasets.items():
start = time.time()
try:
output = orchestrator.run(df, target, train=True)
end = time.time()
results.append({
"dataset": name,
"strategy": output.get("strategy"),
"metrics": output.get("metrics"),
"time": round(end - start, 2),
"error": None
})
except Exception as e:
end = time.time()
results.append({
"dataset": name,
"strategy": None,
"metrics": None,
"time": round(end - start, 2),
"error": str(e)
})
return results
|