Updated sampling function in dist_select

StatMixedML · Aug 25, 2023 · 3f5b5e1 · 3f5b5e1
1 parent 2f0898b
commit 3f5b5e1
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
@@ -594,7 +594,7 @@ def dist_select(self,
                                                         best_dist_sel.param_dict,
                                                         n_targets=best_dist_sel.n_targets,
                                                         rank=best_dist_sel.rank,
-                                                        n_obs=target_expand.shape[0])
+                                                        n_obs=1)
 
             if best_dist["distribution"][0] == "Dirichlet":
                 dist_kwargs = dict(zip(best_dist_sel.distribution_arg_names, [dist_params]))