add word

ml-stat-Sustech · Dec 8, 2023 · 2cdd4aa · 2cdd4aa
1 parent 8063d61
commit 2cdd4aa
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/deepcp/classification/predictor/standard.py b/deepcp/classification/predictor/standard.py
@@ -1,11 +1,11 @@
 import torch
 import numpy as np
 
-from .base import BasePredictor
+from deepcp.classification.predictor.base import BasePredictor
 
 
 class StandardPredictor(BasePredictor):
-    def __init__(self, score_function):0
+    def __init__(self, score_function):
         super().__init__(score_function)
 
 
@@ -17,10 +17,8 @@ def fit(self, x_cal, y_cal, alpha):
 
         self.q_hat = torch.quantile(scores, np.ceil((scores.shape[0] + 1) * (1 - alpha)) / scores.shape[0])
 
-
     def predict(self, x):
         scores = self.score_function.predict(x)
         S = torch.argwhere(scores < self.q_hat).reshape(-1).tolist()
         return S
 
-