Fix titanic tutorial to convert datatypes to float

pytorch / captum

Model interpretability and understanding for PyTorch

https://captum.ai

BSD 3-Clause "New" or "Revised" License

4.96k stars 499 forks source link

Fix titanic tutorial to convert datatypes to float #1331

Closed vivekmig closed 3 months ago

vivekmig commented 3 months ago

Currently, the Titanic tutorial for Captum leads to an error due to mismatched datatypes within the training data. Explicitly converting to a numpy array with floats resolves this issue.

facebook-github-bot commented 3 months ago

@vivekmig has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 3 months ago

@vivekmig merged this pull request in pytorch/captum@d267fd42514c7362787286de51892d7ed0450f4d.