Commit168a40b

committed

little

1 parentf4b7547 commit168a40bCopy full SHA for 168a40b

File tree

-3

lines changed

-3

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,7 @@`
`1`	`1`	`importjson`
`2`	`2`	`importos`
`3`	`3`	`importnumpyasnp`
	`4`	`+fromtqdmimporttqdm`
`4`	`5`	`importargparse`
`5`	`6`
`6`	`7`
`@@ -65,7 +66,7 @@ def get_event_tag(raw_data):`
`65`	`66`	`sent_tag.append(['O'])`
`66`	`67`	`event_argument_tag.append(sent_tag)# because of multi-event,get a tag list for each token`
`67`	`68`
`68`		`-foriinrange(len(raw_data)):`
	`69`	`+foriintqdm(range(len(raw_data)),total=len(raw_data)):`
`69`	`70`	`temp_sent=raw_data[i]`
`70`	`71`	`temp_event_mentions=temp_sent['golden-event-mentions']`
`71`	`72`	`iflen(temp_event_mentions)==0:# skip if none`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,7 @@ def print_count(data):`
`134`	`134`	`event_type= [event["event_type"]foreventincon["golden-event-mentions"]]`
`135`	`135`	`iflen(set(event_type))!=len(event_type):`
`136`	`136`	`multi_same_event_sent_count+=1`
`137`		`-print('Total Sentence: %d;Total Event:%d;Multi-event-sentence: %d;Multi-same-event-sentence: %d'`
	`137`	`+print('[Total Sentence:] %d, [Total Event:]%d,\n [Multi-event-sentence:] %d, [Multi-same-event-sentence:] %d'`
`138`	`138`	`% (len_data,count,multi_event_sent_count,multi_same_event_sent_count))`
`139`	`139`
`140`	`140`
`@@ -172,6 +172,6 @@ def data_split(all_data, rate=[0.8, 0.1, 0.1]):`
`172`	`172`
`173`	`173`	`fornamein ['train','dev','test']:`
`174`	`174`	`save_path=r'output/'+str(args.lang)+'-'+name+'.json'`
`175`		`-print("-"20+name+"-"20)`
	`175`	`+print("-"20+' '+name+' '+"-"20)`
`176`	`176`	`print_count(eval(name))`
`177`	`177`	`save_data(eval(name),save_path)`

Comments

(0)