ai 通用性改动

ai/model_trainer.py

 from abc import ABC
 
-import matplotlib.pyplot as plt
 from lightgbm import LGBMClassifier
 from sklearn import svm
 from sklearn.ensemble import AdaBoostClassifier
 from sklearn.ensemble import RandomForestClassifier, VotingClassifier
-from sklearn.metrics import classification_report, confusion_matrix, ConfusionMatrixDisplay, accuracy_score
+from sklearn.metrics import classification_report, confusion_matrix, accuracy_score
 from sklearn.neighbors import KNeighborsClassifier
 
 from ai.config import LABEL_RANGE