From 8539abe9c57b3f1e955d658d76e810cbca9f093f Mon Sep 17 00:00:00 2001
From: Raffaele Gaetano <raffaele.gaetano@cirad.fr>
Date: Tue, 4 Jul 2023 17:49:12 +0200
Subject: [PATCH] ENH: minor change.

---
 Learning/ObjectBased.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Learning/ObjectBased.py b/Learning/ObjectBased.py
index 893455d..f1dd96c 100644
--- a/Learning/ObjectBased.py
+++ b/Learning/ObjectBased.py
@@ -9,8 +9,10 @@ from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import confusion_matrix, accuracy_score, cohen_kappa_score, precision_recall_fscore_support
 
 class ObjectBasedClassifier:
-    def __init__(self, object_layer, reference_data, time_series_patterns, user_feature_list, ref_class_field='class'):
-        self.obia_base = OBIABase(object_layer, ref_data = reference_data, ref_class_field=ref_class_field)
+    def __init__(self, object_layer, reference_data, time_series_patterns, user_feature_list, ref_class_field='class',
+                 ref_id_field='id'):
+        self.obia_base = OBIABase(object_layer, ref_data=reference_data, ref_class_field=ref_class_field,
+                                  ref_id_field=ref_id_field)
         for ptrn in time_series_patterns:
             lst = sorted(glob.glob(ptrn))
             self.obia_base.add_raster_time_series_for_stats(lst)
-- 
GitLab