Spaces:
Sleeping
Sleeping
File size: 1,167 Bytes
227477c | 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 | from pydantic import BaseModel
from typing import Optional
class InputSchema(BaseModel):
age: int
genre: str
revenu_mensuel: int
statut_marital: str
departement: str
poste: str
nombre_experiences_precedentes: int
nombre_heures_travailless: int
annee_experience_totale: int
annees_dans_l_entreprise: int
annees_dans_le_poste_actuel: int
satisfaction_employee_environnement: int
note_evaluation_precedente: int
niveau_hierarchique_poste: int
satisfaction_employee_nature_travail: int
satisfaction_employee_equipe: int
satisfaction_employee_equilibre_pro_perso: int
note_evaluation_actuelle: int
heure_supplementaires: str
augementation_salaire_precedente: str
nombre_participation_pee: int
nb_formations_suivies: int
nombre_employee_sous_responsabilite: int
distance_domicile_travail: int
niveau_education: int
domaine_etude: str
ayant_enfants: str
frequence_deplacement: str
annees_depuis_la_derniere_promotion: int
annes_sous_responsable_actuel: int
class PredictionOutput(BaseModel):
prediction: int
probability: Optional[float] = None
|