fix multiclass predictions

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

Closed arminaskari closed 1 year ago

arminaskari commented 1 year ago

Fixes a bug in the _predict_cached function for multiclass classification

for multiclass, pred becomes an array instead of a scalar. The offending line is return np.array(preds) which errors if preds (which is a list) contains arrays and scalars (0 when the seq does not belong in the coefsdict)

csinva commented 1 year ago

beautiful