traning_zone/modele_zone/model.py

# Import modèles
from sklearn import  linear_model, naive_bayes, svm
import xgboost
from sklearn.ensemble import GradientBoostingClassifier
from xgboost import XGBClassifier
from sklearn import ensemble

# dictionnaire des modèles
modeles = {"GradientBoostingClassifier": GradientBoostingClassifier(validation_fraction=0.25,random_state=99999),
           "XGBClassifier" : XGBClassifier(random_state=99999),
           "naive_bayes" : naive_bayes.MultinomialNB(),
           "LogisticRegression": linear_model.LogisticRegression(class_weight="balanced"),
           "RandomForestClassifier": ensemble.RandomForestClassifier(class_weight="balanced"),
           "svm" :  svm.SVC(class_weight="balanced")}