Commit 8539abe9 authored by Gaetano Raffaele's avatar Gaetano Raffaele
Browse files

ENH: minor change.

parent 92b16002
No related merge requests found
Showing with 4 additions and 2 deletions
+4 -2
......@@ -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)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment