Fixed col in predict output

ankane · May 6, 2024 · c395e5a · c395e5a
1 parent b46a827
commit c395e5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/prophet/forecaster.rb b/lib/prophet/forecaster.rb
@@ -827,7 +827,7 @@ def predict_seasonal_components(df)
 
       x = seasonal_features.to_numo
       data = {}
-      component_cols.vector_names.each do |component|
+      (component_cols.vector_names - ["col"]).each do |component|
         beta_c =  @params["beta"] * component_cols[component].to_numo
 
         comp = x.dot(beta_c.transpose)