diff --git a/src/python/nimbusml/examples/Image.py b/src/python/nimbusml/examples/Image.py index 08c6aa35..78ee62b6 100644 --- a/src/python/nimbusml/examples/Image.py +++ b/src/python/nimbusml/examples/Image.py @@ -18,10 +18,10 @@ X = data[['Path']] y = data[['Label']] -# define the training pipeline +# define the training pipeline pipeline = Pipeline([ Loader(columns={'ImgPath': 'Path'}), - Resizer(image_width=227, image_height=227, + Resizer(image_width=32, image_height=32, columns={'ImgResize': 'ImgPath'}), PixelExtractor(columns={'ImgPixels': 'ImgResize'}), FastLinearBinaryClassifier(feature='ImgPixels') diff --git a/src/python/nimbusml/examples/examples_from_dataframe/Image_df.py b/src/python/nimbusml/examples/examples_from_dataframe/Image_df.py index 8dd050a0..3cdbb00e 100644 --- a/src/python/nimbusml/examples/examples_from_dataframe/Image_df.py +++ b/src/python/nimbusml/examples/examples_from_dataframe/Image_df.py @@ -18,7 +18,7 @@ # transforms and learners transform_1 = Loader() << 'Path' -transform_2 = Resizer(image_width=227, image_height=227) +transform_2 = Resizer(image_width=32, image_height=32) transform_3 = PixelExtractor() algo = FastLinearBinaryClassifier() << 'Path'