ENH: option to save training base in pkl file.

Learning/ObjectBased.py

 import glob
+import pickle
 
 import numpy as np
 import pandas as pd
@@ -119,6 +120,11 @@ class ObjectBasedClassifier:
                     self.obia_base.populate_map(t, L, c, output_file, compress)
                 prg.update(1)
         return
+    
+    def save_training_base(self, fn):
+        with open(fn, 'wb') as f:
+            pickle.dump(self.training_base, f)
+        return
 
 #TEST CODE
 def run_test(sample_folder):

Workflows/basic.py

@@ -85,6 +85,11 @@ def train_valid_workflow(seg, ts_lst_pkl, d, m_file):
                                 ref_class_field=d['ref_db']['fields'])
 
     obc.gen_k_folds(5, class_field=d['ref_db']['fields'][-1])
+
+    if 'export_training_base' in d['training'].keys() and d['training']['export_training_base'] is True:
+        obc.save_training_base('{}/_side/training_base.pkl'.format(os.path.join(d['output_path'], d['chain_name'])))
+        print('[MORINGA-INFO] : Training base export completed.')
+
     ok = True
 
     for i,cf in enumerate(d['ref_db']['fields']):

Workflows/basic_config.json

@@ -43,6 +43,7 @@
 	],
 	
 	"training": {
+		"export_training_base": false,
 		"classifier": "rf",
 		"parameters": {
 			"n_trees": 400