This was one of the reasons I wrote this, to get feedback and hopefully suggestions about better datasets. I will report back as soon as I find a more appropriate one.

That said I will release another piece showing what I did in the absence of labels for another dataset.