Commit a680cc12 authored by jcyang's avatar jcyang
Browse files

fix metric dimension bug in gaussian_process.py

parent 08f99adf
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