adding a warning for the case of big shift size

MojiFarmanbar · Oct 3, 2023 · d5dc09b · d5dc09b
1 parent 3d8dad4
commit d5dc09b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/conformal_models/transductive_regression.jl b/src/conformal_models/transductive_regression.jl
@@ -858,6 +858,10 @@ function partial_fit(
     ŷₜ = _aggregate(ŷ, aggregate)
     push!(conf_model.scores, @.(conf_model.heuristic(y, ŷₜ))...)
     conf_model.scores = filter(!isnan, conf_model.scores)
+    if shift_size > length(conf_model.scores)
+        @warn "The shift size is bigger than the size of Non-conformity scores"
+        return conf_model.scores
+    end
     conf_model.scores = conf_model.scores[(shift_size + 1):length(conf_model.scores)]
     return conf_model.scores
 end