Processing data for Machine Learning with TensorFlow