Unverified Commit 81b70ad5 authored by Zhenqin Wu's avatar Zhenqin Wu Committed by GitHub
Browse files

Merge pull request #1061 from 0ut0fcontrol/master

fix metric dimension bug in gaussian_process.py for user defined model_class
parents ee8ab952 a680cc12
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -225,8 +225,8 @@ class GaussianProcessHyperparamOpt(HyperparamOpt):
        model.fit(train_dataset, **hyper_parameters)
        model.save()
        evaluator = Evaluator(model, valid_dataset, output_transformers)
        multitask_scores = evaluator.compute_model_performance([metric])
        score = multitask_scores[metric.name]
        multitask_scores = evaluator.compute_model_performance(metric)
        score = multitask_scores[metric[0].name]

      with open(log_file, 'a') as f:
        # Record performances