example 1 changes

krish240574 · Jun 10, 2020 · 4c8d04f · 4c8d04f
1 parent b694609
commit 4c8d04f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/intent_ner_fragment/tasks_file_snips.yml b/examples/intent_ner_fragment/tasks_file_snips.yml
@@ -24,9 +24,9 @@ intent:
     loss_type: CrossEntropyLoss
     task_type: SingleSenClassification
     file_names:
-    - int_snips_train.tsv
-    - int_snips_dev.tsv
-    - int_snips_test.tsv
+    - intent_snips_train.tsv
+    - intent_snips_dev.tsv
+    - intent_snips_test.tsv
 
 
 fragdetect:

diff --git a/utils/tranform_functions.py b/utils/tranform_functions.py
@@ -400,7 +400,7 @@ def create_fragment_detection_tsv(dataDir, readFile, wrtDir, transParamDict, isT
     # saving
     print('writing fragment file for {} at {}'.format(readFile, wrtDir))
 
-    finalDf.to_csv(os.path.join(wrtDir, 'frag_{}.tsv'.format(readFile.split('.')[0])), sep='\t',
+    finalDf.to_csv(os.path.join(wrtDir, 'fragment_{}.tsv'.format(readFile.split('.')[0])), sep='\t',
                 index=False, header=False)
 
 def msmarco_query_type_to_tsv(dataDir, readFile, wrtDir, transParamDict, isTrainFile=False):