def test_validate_GradientBoostingRegressor1(self): fLOG(__file__, self._testMethodName, OutputPrint=__name__ == "__main__") logger = getLogger('skl2onnx') logger.disabled = True verbose = 1 if __name__ == "__main__" else 0 rows = list(enumerate_validated_operator_opsets( verbose, models={"GradientBoostingRegressor"}, opset_min=10, fLOG=fLOG, runtime='python', debug=False)) self.assertGreater(len(rows), 1) max_diff = max(_.get('max_rel_diff_batch', 1e-11) for _ in rows) self.assertLesser(max_diff, 1e-2)
def test_validate_sklearn_store_models(self): fLOG(__file__, self._testMethodName, OutputPrint=__name__ == "__main__") logger = getLogger('skl2onnx') logger.disabled = True rows = list( enumerate_validated_operator_opsets(verbose=0, models={"LinearRegression"}, opset_min=10, store_models=True, fLOG=fLOG)) self.assertNotEmpty(rows) self.assertIn('MODEL', rows[0]) self.assertIn('ONNX', rows[0]) self.assertIsInstance(rows[0]['MODEL'], LinearRegression) oinf = OnnxInference(rows[0]['ONNX']) dot = oinf.to_dot() self.assertIn('LinearRegressor', dot)
def test_rt_grid_search_cv(self): fLOG(__file__, self._testMethodName, OutputPrint=__name__ == "__main__") verbose = 1 if __name__ == "__main__" else 0 buffer = [] def myprint(*args, **kwargs): buffer.append(" ".join(map(str, args))) debug = True rows = list( enumerate_validated_operator_opsets( verbose, models={"GridSearchCV"}, fLOG=myprint, runtime='python', debug=debug, filter_exp=lambda m, p: "64" not in p)) self.assertGreater(len(rows), 1) self.assertGreater(len(buffer), 1 if debug else 0)
def test_rt_isonotic_regression(self): fLOG(__file__, self._testMethodName, OutputPrint=__name__ == "__main__") logger = getLogger('skl2onnx') logger.disabled = True verbose = 1 if __name__ == "__main__" else 0 buffer = [] def myprint(*args, **kwargs): buffer.append(" ".join(map(str, args))) debug = False rows = list( enumerate_validated_operator_opsets( verbose, models={"IsotonicRegression"}, fLOG=myprint, runtime='python', debug=debug, filter_exp=lambda m, p: "64" not in p)) self.assertGreater(len(rows), 1) self.assertGreater(len(buffer), 1 if debug else 0)