Merge pull request #139 from scikit-learn-contrib/dev

doc examples patched

Author: JulienRoussel77

examples/tutorials/plot_tuto_categorical.py

@@ -21,7 +21,7 @@
 # ---------------------------------------------------------------
 # We get the data and focus on the explanatory variables
 df = data.get_data("Titanic")
-df = df.drop(columns=["survived"])
+df = df.drop(columns=["Survived"])
 
 # %%
 # 2. Mixed type imputation methods

qolmat/utils/data.py

@@ -324,7 +324,7 @@ def add_holes(df: pd.DataFrame, ratio_masked: float, mean_size: int) -> pd.DataF
     pd.DataFrame
         dataframe with missing values
     """
-    groups = df.index.names.difference(["datetime", "date", "index"])
+    groups = df.index.names.difference(["datetime", "date", "index", None])
     if groups != []:
         generator = missing_patterns.GeometricHoleGenerator(
             1, ratio_masked=ratio_masked, subset=df.columns, groups=groups