finished training using pandas