Doctorado-ML/bayesclass
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/bayesclass.git synced 2025-08-17 16:45:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add messages to check_integrity

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-03-23 22:10:03 +01:00
parent ea8c5b805e
commit 7b66097728
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bayesclass/clfs.py
View File

@@ -532,7 +532,7 @@ class Proposal:
kwargs = self.update_kwargs(y, kwargs) kwargs = self.update_kwargs(y, kwargs)
# Build the model # Build the model
super(self.class_type, self.estimator).fit(self.Xd, y, **kwargs) super(self.class_type, self.estimator).fit(self.Xd, y, **kwargs)
self.check_integrity("f", self.Xd) self.check_integrity("fit", self.Xd)
# Local discretization based on the model # Local discretization based on the model
features = kwargs["features"] features = kwargs["features"]
# assign indices to feature names # assign indices to feature names
@@ -543,7 +543,7 @@ class Proposal:
super(self.class_type, self.estimator).fit(self.Xd, y, **kwargs) super(self.class_type, self.estimator).fit(self.Xd, y, **kwargs)
def predict(self, X): def predict(self, X):
self.check_integrity("p", self.discretizer.transform(X)) self.check_integrity("predict", self.discretizer.transform(X))
return super(self.class_type, self.estimator).predict( return super(self.class_type, self.estimator).predict(
self.discretizer.transform(X) self.discretizer.transform(X)
) )