search

csinva / imodelsX

Scikit-learn friendly library to interpret, and prompt-engineer text datasets using large language models.
https://csinva.io/imodelsX/
MIT License
161 stars 26 forks source link

Class scores #9

Closed valeman closed 6 months ago

valeman commented 6 months ago

How can one get class scores from KAN_Classifier? predict_proba returns 0 and 1 instead of class scores it seems.

csinva commented 6 months ago

Thanks for the question!

Sorry about that, original version didn't implement this properly, just pushed the latest version to pypi so predict_proba should work as expected (see the notebook) for the working example.

To see the change, either run pip install --upgrade imodelsx or install from source.

valeman commented 6 months ago

Hi @csinva thats fantastic, thank you.