Update hyperopt.py
This commit is contained in:
parent
1a573d57b9
commit
d2a5448305
@ -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)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user