From 2811cd1adbea3f2027d30e8752674615ecfbaa0b Mon Sep 17 00:00:00 2001
From: Nirmal Borah <nirmalborah93@gmail.com>
Date: Fri, 14 Sep 2018 13:49:04 +0530
Subject: [PATCH] Some more lines are added to the exporter

---
 nyoka/lgbm/lgb_to_pmml.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/nyoka/lgbm/lgb_to_pmml.py b/nyoka/lgbm/lgb_to_pmml.py
index 64cd631..5ca81e2 100644
--- a/nyoka/lgbm/lgb_to_pmml.py
+++ b/nyoka/lgbm/lgb_to_pmml.py
@@ -66,6 +66,13 @@ def lgb_to_pmml(pipeline, col_names, target_name, pmml_f_name='from_lgbm.pmml'):
             **PMML_kwargs
         )
         pmml.export(outfile=open(pmml_f_name, "w"), level=0)
+        read_file = open(pmml_f_name, 'r')
+        lines = read_file.readlines()
+        lines.insert(1,'<!--(Comment generated by ADAPA) PMML processed by ADAPA (Version : 4.3)-->\n')
+        read_file.close()
+        read_file = open(pmml_f_name, 'w')
+        read_file.writelines(lines)
+        read_file.close()
 
 def get_PMML_kwargs(model, derived_col_names, col_names, target_name, mining_imp_val,categoric_values):
     """