Update validate_problems.py

sdpython · Jul 22, 2019 · 8cb4c4c · 8cb4c4c
1 parent 85c8a43
commit 8cb4c4c
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/mlprodict/onnxrt/validate_problems.py b/mlprodict/onnxrt/validate_problems.py
@@ -320,13 +320,15 @@ def find_suitable_problem(model):
     def _internal(model):
         # Exceptions
         if model in {GaussianProcessRegressor}:
-            return ['~b-reg-NF-64', '~m-reg-NF-64',
-                    '~b-reg-NF-cov-64', '~m-reg-NF-cov-64',
-                    '~b-reg-NF-std-64', '~m-reg-NF-std-64',
-                    '~b-reg-NSV-64', '~m-reg-NSV-64',
-                    '~b-reg-cov-64', '~m-reg-cov-64',
-                    '~b-reg-std-NSV-64', '~m-reg-std-NSV-64',
-                    'b-reg', '~b-reg-64', 'm-reg']
+            # m-reg causes MemoryError on some machine.
+            return ['~b-reg-NF-64',  # '~m-reg-NF-64',
+                    '~b-reg-NF-cov-64',  # '~m-reg-NF-cov-64',
+                    '~b-reg-NF-std-64',  # '~m-reg-NF-std-64',
+                    '~b-reg-NSV-64',  # '~m-reg-NSV-64',
+                    '~b-reg-cov-64',  # '~m-reg-cov-64',
+                    '~b-reg-std-NSV-64',  # '~m-reg-std-NSV-64',
+                    'b-reg', '~b-reg-64',  # 'm-reg'
+                    ]
 
         if model in {BaggingClassifier, BernoulliNB, CalibratedClassifierCV,
                      ComplementNB, GaussianNB, GaussianProcessClassifier,