Under the load_data.py, add scripts for data preprocessing: - flatten the matrix for training and test set - scale the data if needed - convert output variable to be categorical if needed