utils.py [output_mode = regression]

[vr25]

Hi,

I am trying to predict a score (float) using regression and I was wondering what changes need to be made in lines 254 and 258 in utils.py

Thanks!

[vr25]

Also, since it uses BinaryClassificationProcessor, it creates the logits tensor with size of [10,2] (say, my input file has 10 texts). After view(-1), logits becomes size=20 but the labels are still size=10. Hence, MSELoss() returns the following size mismatch error:

Thanks!

[ThilinaRajapakse]

Set this in the args dict.

'output_mode': 'regression'

[vr25]

On setting:
'output_mode': 'regression'
it still gives the following error: