Skip to content

Pytorch output support in lime_tabular.py#697

Open
UrjaPawar wants to merge 2 commits intomarcotcr:masterfrom
UrjaPawar:master
Open

Pytorch output support in lime_tabular.py#697
UrjaPawar wants to merge 2 commits intomarcotcr:masterfrom
UrjaPawar:master

Conversation

@UrjaPawar
Copy link

If 'yss' is output of a pytorch model, then this check is necessary in order to run a pytorch model trained on tabular dataset: if not isinstance(yss, np.ndarray):
yss = yss.detach().numpy()

If 'yss' is output of a pytorch model, then this check is necessary in order to run a pytorch model trained on tabular dataset:
if not isinstance(yss, np.ndarray):
     yss = yss.detach().numpy()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant