diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py
index ce028512357f2..0c8c60d2a57a2 100644
--- a/python/pyspark/ml/classification.py
+++ b/python/pyspark/ml/classification.py
@@ -17,6 +17,8 @@
 
 import operator
 import sys
+import uuid
+
 from multiprocessing.pool import ThreadPool
 
 from pyspark import since, keyword_only