Skip to content

Commit 0a8f441

Browse files
committed
add decision function
1 parent c099b28 commit 0a8f441

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

Diff for: afc_imbalanced_learning/afc.py

+7
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,13 @@ def predict_proba(self, X):
6767
X, self.X_train, computed_kernel, self.support_vectors, self.tau_squareds
6868
)
6969
return self.svm.predict_proba(computed_conformal_transform_kernel)
70+
71+
def decision_function(self, X):
72+
computed_kernel = self.kernel(X, self.X_train)
73+
computed_conformal_transform_kernel = conformal_transform_kernel(
74+
X, self.X_train, computed_kernel, self.support_vectors, self.tau_squareds
75+
)
76+
return self.svm.decision_function(computed_conformal_transform_kernel)
7077

7178
def extract_separate_support_vectors(self):
7279
support_vectors = self.X_train[self.svm.support_]

Diff for: release-notes.txt

-5
This file was deleted.

0 commit comments

Comments
 (0)