cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
http://contextualized.ml/
GNU General Public License v3.0
65 stars 9 forks source link

add demo for Classifier, fix predict_ shapes for classifier #88

Closed blengerich closed 2 years ago

blengerich commented 2 years ago

predict_proba now predicts an array of shape N x O x 2, just like sklearn interface.

Also added a new demo notebook specifically focused on classification, and removed an extraneous print statement.