finished converting to dataframe and splitting into train, validate and test sets