A library that incorporates state-of-the-art explainers for text-based machine learning models and visualizes the result with a built-in dashboard.
409
stars
68
forks
source link
Running text_classification_introspective_rationale_explainer.ipynb with multi-class dataset #222
Open
HenryLiuPI opened 10 months ago
I'm getting RuntimeError: multi-target not supported at ..\aten\src\THNN/generic/ClassNLLCriterion.c:20 while trying to feed multi-class dataset.
My training data looks like this. I'm using one-hot format as label.![image](https://github.com/interpretml/interpret-text/assets/48251472/b102f1cf-003e-4a87-8e8f-36a6c7ab7164)
I'm getting the error on explainer.fit(df_train, df_test)![image](https://github.com/interpretml/interpret-text/assets/48251472/592bfadf-5451-4e05-a0a4-14cbf1580ff0)
The dataset is a multi-class dataset and has 8 labels in total. Each sample may contain multi labels. What format should I use to represent the label?