Update README.md

README.md

@@ -36,11 +36,10 @@ from sklearn.pipeline import Pipeline
 model = joblib.load("iris_svm.joblib")
 
 column_transformer_pipeline = ColumnTransformer([
-                ("loading_missing_value_imputer", SimpleImputer(strategy="mean"), ["loading"]),
-                ("numerical_missing_value_imputer", SimpleImputer(strategy="mean"), list(df.columns[df.dtypes == 'float64'])),
-                ("attribute_0_encoder", OneHotEncoder(categories = "auto"), ["attribute_0"]),
-                ("attribute_1_encoder", OneHotEncoder(categories = "auto"), ["attribute_1"]),
-                ("product_code_encoder", OneHotEncoder(categories = "auto"), ["product_code"])])
+                ("SepalLengthCm", SimpleImputer(strategy="mean"), ["SepalLengthCm"]),
+                ("SepalWidthCm", SimpleImputer(strategy="mean"), ["SepalWidthCm"]),
+                ("PetalLengthCm", SimpleImputer(strategy="mean"), ["PetalLengthCm"]),
+                ("PetalWidthCm", SimpleImputer(strategy="mean"), ["PetalWidthCm"])])
 import json
 pipeline = Pipeline([
     ('transformation', column_transformer_pipeline),