guezoloic/millesima-ai-engine
1
0
Fork 0
mirror of https://github.com/guezoloic/millesima-ai-engine.git synced 2026-03-28 18:03:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

ajout : remplacer appellation par les colonnes indicatrices

Browse Source
This commit is contained in:
Chahrazad650
2026-03-03 03:26:58 +01:00
parent b0eb5df07e
commit 06097c257e
2 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
cleaning.py
View File

@@ -74,3 +74,18 @@ def fill_missing_scores(df: DataFrame) -> DataFrame:
df_copy = df_copy.drop(columns=temp_cols)
return df_copy
def encode_appellation(df: DataFrame, column: str = "Appellation") -> DataFrame:
"""
Remplace la colonne 'Appellation' par des colonnes indicatrices
"""
df_copy = df.copy()
appellations = df_copy[column].astype(str).str.strip()
appellation_dummies = pd.get_dummies(appellations)
df_copy = df_copy.drop(columns=[column])
return df_copy.join(appellation_dummies)

6
main.py
View File

@@ -59,6 +59,12 @@ def main() -> None:
print("\n=== Après remplissage des notes manquantes ===")
display_info(df_missing_scores)
df_ready = encode_appellation(df_missing_scores)
save_csv(df_ready, "donnee_ready.csv")
print("\n=== Après remplacer la colonne 'Appellation' par des colonnes indicatrices ===")
display_info(df_ready)
print(df_ready.filter(like="App_").any().head())
if __name__ == "__main__":
try:
main()