Fixes TheAlgorithms#6551

Moddy2024 · web-flow · commit 79603f77b293 · 2022-10-15T00:45:30.000+05:30
diff --git a/machine_learning/xgboostclassifier.py b/machine_learning/xgboostclassifier.py
@@ -4,6 +4,7 @@
 from sklearn.metrics import plot_confusion_matrix
 from sklearn.model_selection import train_test_split
 from xgboost import XGBClassifier
+import doctest
 
 
 def data_handling(data: dict) -> tuple:
