Skip to content

Running text_classification_introspective_rationale_explainer.ipynb with multi-class dataset #222

@HenryLiuPI

Description

@HenryLiuPI

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

I'm getting the error on explainer.fit(df_train, df_test)
image

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions