I had to make the dataset again because I didn't save it last time, now I did some cleaning and change the settings for the new dataset to see if can make it work better.