From e466fb5516dd107b52b71e2663ab47a33ecd8375 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexander=20M=C3=A4rz?= Date: Wed, 6 Mar 2024 10:18:46 +0100 Subject: [PATCH] Update initialization of eval_sets --- lightgbmlss/model.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/lightgbmlss/model.py b/lightgbmlss/model.py index f966a87..33896ce 100644 --- a/lightgbmlss/model.py +++ b/lightgbmlss/model.py @@ -170,7 +170,7 @@ def train(self, self.set_init_score(train_set) if valid_sets is not None: - valid_sets = self.set_valid_margin(valid_sets, self.start_values) + valid_sets = self.set_valid_margin(valid_sets) self.booster = lgb.train(params, train_set, @@ -551,7 +551,6 @@ def expectile_plot(self, def set_valid_margin(self, valid_sets: list, - start_values: np.ndarray ) -> list: """ Function that sets the base margin for the validation set. @@ -559,16 +558,12 @@ def set_valid_margin(self, Arguments --------- valid_sets : list - List of tuples containing the train and evaluation set. - valid_names: list - List of tuples containing the name of train and evaluation set. - start_values : np.ndarray - Array containing the start values for the distributional parameters. + List of tuples containing the evaluation set(s). Returns ------- valid_sets : list - List of tuples containing the train and evaluation set. + List of tuples containing the evaluation set(s). """ for valid_set in valid_sets: self.set_init_score(valid_set)