Added 'EstimatorProxy.classes_' attribute

jpmml · Mar 3, 2024 · e766c42 · e766c42
1 parent 8761994
commit e766c42
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sklearn2pmml/__init__.py b/sklearn2pmml/__init__.py
@@ -76,6 +76,10 @@ def _copy_attrs(self):
 			if hasattr(self.estimator, attr_name):
 				setattr(self, attr_name, getattr(self.estimator, attr_name))
 
+	@property
+	def classes_(self):
+		return self.estimator.classes_
+
 	def fit(self, X, y = None, **fit_params):
 		self.estimator.fit(X, y, **fit_params)
 		self._copy_attrs()