guezoloic/archived-millesima-projetS6
1
0
Fork 0
mirror of https://github.com/guezoloic/millesima_projetS6.git synced 2026-03-28 19:13:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'jalon2-loic' of https://github.com/guezoloic/millesima_projetS6

Browse Source
This commit is contained in:
Loïc GUEZO
2026-03-06 21:36:48 +01:00
parent bcacd7a915 4b3c3c26e8
commit 6fbb36ea37
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/cleaning.py
View File

@@ -14,11 +14,11 @@ def path_filename(filename: str) -> str:
class Cleaning: class Cleaning:
def __init__(self, filename) -> None: def __init__(self, filename) -> None:
self._vins: DataFrame = read_csv(filename) self._vins: DataFrame = read_csv(filename)
# # créer la liste de tout les scores
self.SCORE_COLS: list[str] = [ self.SCORE_COLS: list[str] = [
c for c in self._vins.columns if c not in ["Appellation", "Prix"] c for c in self._vins.columns if c not in ["Appellation", "Prix"]
] ]
# # transforme tout les colonnes score en numérique
for col in self.SCORE_COLS: for col in self.SCORE_COLS:
self._vins[col] = to_numeric(self._vins[col], errors="coerce") self._vins[col] = to_numeric(self._vins[col], errors="coerce")
@@ -87,7 +87,7 @@ class Cleaning:
Remplace la colonne 'Appellation' par des colonnes indicatrices Remplace la colonne 'Appellation' par des colonnes indicatrices
""" """
appellations = self._vins[column].astype(str).str.strip() appellations = self._vins[column].astype(str).str.strip()
appellation_dummies = get_dummies(appellations) appellation_dummies = get_dummies(appellations, prefix="App")
self._vins = self._vins.drop(columns=[column]) self._vins = self._vins.drop(columns=[column])
self._vins = self._vins.join(appellation_dummies) self._vins = self._vins.join(appellation_dummies)
return self return self

6
tests/test_cleaning.py
View File

@@ -62,6 +62,6 @@ def test_encode_appellation(cleaning_raw: Cleaning):
.encode_appellation() .encode_appellation()
.getVins() .getVins()
) )
assert "Appellation" not in out.columns assert "App_Appellation" not in out.columns
assert "Pauillac" in out.columns assert "App_Pauillac" in out.columns
assert int(out.loc[0, "Pauillac"]) == 1 assert int(out.loc[0, "App_Pauillac"]) == 1