csinva / imodelsX

Scikit-learn friendly library to interpret, and prompt-engineer text datasets using large language models.

Home Page:https://csinva.io/imodelsX/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Class scores

valeman opened this issue · comments

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

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.

Hi @csinva thats fantastic, thank you.