Decouple train and eval dataset

[yuki-97]

As title.

Train and eval dataset are coupled for now, which means we need write the same logic twice for train and eval when we add support for new dataset, so it's good to decouple them.