Update hyperopt.py

This commit is contained in:
Italo 2022-03-11 17:38:32 +00:00
parent 1a573d57b9
commit d2a5448305

View File

@ -486,6 +486,7 @@ class Hyperopt:
if res.models and hasattr(res.models[-1], "kernel_"): if res.models and hasattr(res.models[-1], "kernel_"):
print(f'kernel: {res.models[-1].kernel_}') print(f'kernel: {res.models[-1].kernel_}')
print(datetime.now())
# Calculate progressbar outputs # Calculate progressbar outputs
for j, val in enumerate(f_val): for j, val in enumerate(f_val):
@ -542,7 +543,6 @@ class Hyperopt:
model.criterion = 'squared_error' model.criterion = 'squared_error'
r2 = cross_val_score(model, X=res.x_iters, y=res.func_vals, scoring='r2', cv=5, n_jobs=jobs).mean() r2 = cross_val_score(model, X=res.x_iters, y=res.func_vals, scoring='r2', cv=5, n_jobs=jobs).mean()
print(f'R2: {r2}')
r2 = r2 if r2 > -5 else -5 r2 = r2 if r2 > -5 else -5
self.r2_list.append(r2) self.r2_list.append(r2)