Skip to content

Commit

Permalink
Merge pull request #354 from piotr-semenov/fix/individual_ensemble_pr…
Browse files Browse the repository at this point in the history
…edictions__multiclass

For any multiclass task, `make_predictions` fails if option --individual_ensemble_predictions is on.
  • Loading branch information
oscarwumit committed Feb 4, 2023
2 parents c371063 + e742803 commit ad97341
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion chemprop/uncertainty/uncertainty_estimator.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import numpy as np
from typing import Iterator, List

from chemprop.data import MoleculeDataset, StandardScaler
Expand Down Expand Up @@ -62,4 +63,4 @@ def individual_predictions(self):
"""
Return separate predictions made by each individual model in an ensemble of models.
"""
return self.predictor.get_individual_preds()
return np.asarray(self.predictor.get_individual_preds())

0 comments on commit ad97341

Please sign in to comment.