Logging F1 Score and RMSE

[amahonycxnpl]

Hi!

Thanks for sharing the code.

What changes would need to be made to the code to output (and log) the F1 Score and RMSE for TabBert? This way I would be able to reproduce the results in the paper.

Currently, training a model just logs the following:
{'train_runtime', 'train_samples_per_second', 'train_steps_per_second', 'train_loss', 'epoch'}

Any help would be much appreciated - thanks.