Issue running example

[hcorbaz]

Hello,

This project looks really interesting, but I encountered some error while running the example. When running the file titanic_missingness_pipeline.py, this error message is happening RuntimeError: Predictions and targets are expected to have the same shape, but got torch.Size([43]) and torch.Size([177]).

Thanks a lot!

[ACoeruleus]

Hi,

thanks for bringing this up, I had the same mistake. For me the code runs when I unindent twice the lines 347-357 in 'multimodn.py' (creation of the confusion matrices).

Not sure if this is really what is needed though...

Best, AC

[imannajwa-masrun]

Hi,

thanks for bringing this up, I had the same mistake. For me the code runs when I unindent twice the lines 347-357 in 'multimodn.py' (creation of the confusion matrices).

Not sure if this is really what is needed though...

Best, AC

Had the same issue and this worked! Thank you.

[hcorbaz]

Hello,

Thanks a lot for your answer! It indeed fixed the training, but the validation looks wrong: it returns 0.5 for each epoch and the plots are empty. This confusion matrix is confusing me.
Thanks again for your help!