Update worst examples generation to use training data instead (#6158)
Update worst examples generation to use training data instead (#6158)

* Update train.py

* Update train.py
mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow