On prediction.py, use the validate_on_data to create the funtion _states_data,...

On prediction.py, use the validate_on_data to create the funtion _states_data, wich return the states representation on the same way as the translation funtion