ppdebreuck / modnet

MODNet: a framework for machine learning materials properties
MIT License
80 stars 34 forks source link

Hard and soft voting for `predict` of `EnsembleMODNetModel` #245

Open kaueltzen opened 3 days ago

kaueltzen commented 3 days ago

Resolves #225 .

@ppdebreuck let me know if you need anything else(:

Also I want to note that with the current changes, the shapes of df_mean and df_std differ for soft voting (df_std contains stds on class probability predictions, df_mean integer class labels), but I don't know if anything else would make more sense.