Added MODEL_TYPE param, which is needed to represent the test set.