File size: 2,289 Bytes
15524ff
23d4613
15524ff
 
 
23d4613
15524ff
 
23d4613
15524ff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23d4613
15524ff
23d4613
15524ff
 
 
 
23d4613
15524ff
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from App.database import Base, SQLALCHEMY_AVAILABLE

if SQLALCHEMY_AVAILABLE: 
    from sqlalchemy import Column, Integer, String, Float, Boolean, DateTime, ForeignKey 
    from sqlalchemy.sql import func

    class Input(Base):
        __tablename__ = "inputs"

        id = Column(Integer, primary_key=True, index=True)
        genre = Column(String)
        statut_marital = Column(String)
        departement = Column(String)
        poste = Column(String)
        domaine_etude = Column(String)
        frequence_deplacement = Column(String)
        heure_supplementaires = Column(Boolean)
        evolution_cat_evol = Column(String)
        categorie_employe = Column(String)
        satisfaction_employee_nature_travail = Column(Integer)
        nombre_participation_pee = Column(Integer)
        ecart_note_evaluation = Column(Integer)
        revenu_mensuel = Column(Integer)
        distance_domicile_travail = Column(Integer)
        satisfaction_globale = Column(Float)
        niveau_education = Column(Integer)
        note_evaluation_actuelle = Column(Integer)
        satisfaction_employee_equipe = Column(Integer)
        age = Column(Integer)
        revenu_par_annee_experience_interne = Column(Integer)
        satisfaction_employee_equilibre_pro_perso = Column(Integer)
        nombre_experiences_precedentes = Column(Integer)
        annees_dans_l_entreprise = Column(Integer)
        nb_formations_suivies = Column(Integer)
        revenu_par_annee_experience_totale = Column(Integer)
        ratio_sans_promotion = Column(Integer)
        satisfaction_employee_environnement = Column(Integer)
        exp_hors_entreprise = Column(Integer)
        mobilite_promotion = Column(Integer)
        annees_depuis_la_derniere_promotion = Column(Integer)

        created_at = Column(DateTime(timezone=True), server_default=func.now())

    class Predictions(Base):
        __tablename__ = "predictions"
        id = Column(Integer, primary_key=True, index=True)
        input_id = Column(Integer, ForeignKey("inputs.id"))

        prediction_label = Column(String)
        prediction_proba = Column(Float)
        model_version = Column(String)

        created_at = Column(DateTime(timezone=True), server_default=func.now())
else: 
    Input = None 
    Predictions = None