UPD: simply code in crf score

scikit-learn-contrib · Jul 19, 2023 · ffbdeee · ffbdeee
1 parent aa3431d
commit ffbdeee
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/mapie/conformity_scores/residual_conformity_scores.py b/mapie/conformity_scores/residual_conformity_scores.py
@@ -392,12 +392,9 @@ def get_signed_conformity_scores(
  )
 
  # reconstruct array with nan and conformity scores
- complete_signed_cs = np.zeros_like(y_pred)
+ complete_signed_cs = np.zeros_like(y_pred, dtype=float)
  complete_signed_cs[cal_indexes] = signed_conformity_scores
- complete_signed_cs[train_indexes] = np.full(
- (train_indexes.shape[0],),
- np.nan
- )
+ complete_signed_cs[train_indexes] = np.nan
  return signed_conformity_scores
 
  def get_estimation_distribution(