Tensorflow One Hot Encoding Python Dataframe