somepago / saint

The official PyTorch implementation of recent paper - SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Training
Apache License 2.0
402 stars 63 forks source link

Errors about dividing by zero #13

Open LeCongThuong opened 2 years ago

LeCongThuong commented 2 years ago

It may help if your add minor number like 1e-8, 1e-9 at thís line (link) to make your code more robust.