Fix mushroom dataset loading and training

It works now, but it's pretty hacky. Also fixed an issue with an if condition
in main.cu.