Fix bug when loading dataset which is going to be predicted.
Add Model abstract class.
Add command arguments for training and prediction.