Changed counter for samples

StatMixedML · Aug 8, 2023 · 039a0ac · 039a0ac
1 parent d28926d
commit 039a0ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xgboostlss/distributions/multivariate_distribution_utils.py b/xgboostlss/distributions/multivariate_distribution_utils.py
@@ -323,7 +323,7 @@ def draw_samples(self,
         for i in range(self.n_targets):
             target_df = pd.DataFrame.from_dict({"target": [f"y{i + 1}" for _ in range(dist_samples.shape[1])]})
             df_samples = pd.DataFrame(dist_samples[i, :])
-            df_samples.columns = [str("y_sample") + str(i+1) for i in range(n_samples)]
+            df_samples.columns = [str("y_sample") + str(i) for i in range(n_samples)]
             samples_list.append(pd.concat([target_df, df_samples], axis=1))
 
         samples_df = pd.concat(samples_list, axis=0).reset_index(drop=True)