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

Issues with datasets #22

Open YishGene opened 1 year ago

YishGene commented 1 year ago

Hi,

First of all - very impressive project and repository! Chapeau to you all.

I'm trying to generate some of the results, having issues with, e.g., HTRU2. I searched openml datasets and reached dataset id 43377, but this doesn't load with your code (y values are None).

Maybe I'm looking in the wrong place, but - could you provide a list of opeml dataset ids to recreate the results in your paper?