Initial

README.md

@@ -126,7 +126,7 @@ The inference result will be saved in ```./saved_models/Fine_Tuned_Model/result.
 
 ## 5. Fine-Tuning (**Optional**)
 
-**Run fine-tuning with:**
+- **Run fine-tuning with:**
 ```
 $ bash run_fine_tuning.sh
 ```

Scripts/UnixCoder/run_one_model.py

@@ -77,8 +77,8 @@ def read_examples_no_bracket(filename, is_function_test):
                     break
             line = line.strip()
             js = json.loads(line)
-            # if len(js['Stmt_token']) > 300:
-            #     continue
+            if idx > 1000:
+                break
             if js["Stmt"].strip()[0] == "}":
                 continue
             if js["Value"].strip().lower() == "nothing" and '#' in js['FIR']:
@@ -152,8 +152,8 @@ def read_examples(filename, is_function_test):
                     break
             line = line.strip()
             js = json.loads(line)
-            # if len(js['Stmt_token']) > 300:
-            #     continue
+            if idx > 3000:
+                break
             if 'idx' not in js:
                 js['idx'] = idx
             code = ' '.join(js['FIR_token']).replace('\n', ' ')