@@ -22,9 +22,9 @@ def replace_location_entities(dataset_text, dataset_mms):
22
22
'Notbremse' , 'Reservierungen' ,'IC 2313' ,'Sonderzug' ,'Rhein' }
23
23
24
24
25
- for folder_name , file in dataset_text .items ():
25
+ for folder_name , file_content in dataset_text .items ():
26
26
file_with_metadata = []
27
- for (start_time , end_time , sentence , number ) in file :
27
+ for (start_time , end_time , sentence , number ) in file_content :
28
28
sentences_to_analyze = sentence .strip ().translate ({ord (i ): None for i in "„“" })
29
29
# sentences_to_analyze = sentence.strip().replace("„", "").replace("“", "")
30
30
doc = nlp (sentences_to_analyze )
@@ -50,9 +50,9 @@ def replace_location_entities(dataset_text, dataset_mms):
50
50
result_text = {}
51
51
result_mms = {}
52
52
53
- for folder_name , file in dataset_text_with_metadata .items ():
53
+ for folder_name , file_content in dataset_text_with_metadata .items ():
54
54
location_counts = {}
55
- for line_number , (start_time , end_time , sentence , number , entities ) in enumerate (file ):
55
+ for line_number , (start_time , end_time , sentence , number , entities ) in enumerate (file_content ):
56
56
for (text , label ) in entities :
57
57
if label == 'LOC' and text not in excludes :
58
58
if 'Hauptbahnhof' in text :
@@ -71,7 +71,7 @@ def replace_location_entities(dataset_text, dataset_mms):
71
71
72
72
73
73
new_text_data = []
74
- for (start_time , end_time , sentence , number , entities ) in file :
74
+ for (start_time , end_time , sentence , number , entities ) in file_content :
75
75
sentence , _ = replace_multiple (sentence , mapping )
76
76
new_text_data .append ((start_time , end_time , sentence , number ))
77
77
result_text [folder_name ] = new_text_data
0 commit comments