NN and RF was using 80% of train data