From 9955d6a859616fcc80fb7eea0134c375bcb1330f Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 18 Jan 2022 16:35:00 +0100
Subject: optionally export xgboost model

---
 lib/parameters.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'lib')

diff --git a/lib/parameters.py b/lib/parameters.py
index 3157c91..50fefbe 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -965,6 +965,9 @@ class ModelAttribute:
             self.model_function = df.SKLearnRegressionFunction(
                 np.mean(data), xgb, category_to_index, ignore_index
             )
+            output_filename = os.getenv("DFATOOL_XGB_DUMP_MODEL", None)
+            if output_filename:
+                xgb.dump_model(output_filename, dump_format="json", with_stats=True)
             return
 
         if loss_ignore_scalar and not with_function_leaves:
-- 
cgit v1.2.3