change method name by adding prefix 'MIT', 'SAT', 'NLP'

Author: Naplues

main.py

@@ -1,6 +1,7 @@
 # -*- coding:utf-8 -*-
 import warnings
 import sys
+import time
 from src.models.tools import *
 from src.models.explain import *
 from src.models.glance import *
@@ -11,27 +12,41 @@
 simplefilter(action='ignore', category=FutureWarning)
 
 # The model name and its corresponding python class implementation
-MODEL_DICT = {'TMI-LR': TMI_LR, 'TMI-SVM': TMI_SVM, 'TMI-MNB': TMI_MNB, 'TMI-DT': TMI_DT, 'TMI-RF': TMI_RF,
-              'LineDP': LineDP,
-              'PMD': PMD, 'CheckStyle': CheckStyle,
-              'NGram': NGram, 'NGram-C': NGram_C,
-              'Glance-EA': Glance_EA, 'Glance-MD': Glance_MD, 'Glance-LR': Glance_LR, 'Glance-EA-SATD': Glance_EA_SATD,
+MODEL_DICT = {'MIT-TMI-LR': TMI_LR, 'MIT-TMI-SVM': TMI_SVM, 'MIT-TMI-MNB': TMI_MNB, 'MIT-TMI-DT': TMI_DT,
+              'MIT-TMI-RF': TMI_RF, 'MIT-LineDP': LineDP,
+              'SAT-PMD': PMD, 'SAT-CheckStyle': CheckStyle,
+              'NLP-NGram': NGram, 'NLP-NGram-C': NGram_C,
+              'Glance-EA': Glance_EA, 'Glance-MD': Glance_MD, 'Glance-LR': Glance_LR,
               }
 
 
 # ========================= Run RQ1 experiments =================================
 def run_cross_release_predict(prediction_model):
+    # time
+    release_name, build_time_list, pred_time_list = [], [], []
     for project, releases in get_project_releases_dict().items():
         for i in range(len(releases) - 1):
             # 1. Loading data. train data index = i, test data index = i + 1
             print(f'========== {prediction_model.model_name} CR PREDICTION for {releases[i + 1]} ================'[:60])
+            # ####### Build time #######
+            t_start = time.time()
             model = prediction_model(releases[i], releases[i + 1])
+            t_end = time.time()
+            build_time_list.append(t_end - t_start)
 
+            # ####### Pred time #######
+            t_start = time.time()
             model.file_level_prediction()
-            model.analyze_file_level_result()
-
+            # model.analyze_file_level_result()
             model.line_level_prediction()
-            model.analyze_line_level_result()
+            # model.analyze_line_level_result()
+            t_end = time.time()
+            pred_time_list.append(t_end - t_start)
+            release_name.append(releases[i + 1])
+
+            data = {'release_name': release_name, 'build_time': build_time_list, 'pred_time': pred_time_list}
+            data = pd.DataFrame(data, columns=['release_name', 'build_time', 'pred_time'])
+            data.to_csv(model.time_file, index=False)
 
 
 def run_default():
@@ -41,10 +56,11 @@ def run_default():
     # ======= NLP-based approaches ======= NGram, NGram_C
     # ======= Glance-XX approaches ======= Glance_MD, Glance_EA, Glance_LR
 
-    # run_cross_release_predict(Glance_LR)
+    run_cross_release_predict(Glance_LR)
     # run_cross_release_predict(Glance_EA)
     # run_cross_release_predict(Glance_MD)
-    run_cross_release_predict(LineDP)
+    # run_cross_release_predict(LineDP)
+    pass
 
 def parse_args():
     # If there is no additional parameters in the command line, run the default models.

src/models/__pycache__/base_model.cpython-38.pyc

@@ -27,6 +27,7 @@ def __init__(self, train_release: str = '', test_release: str = '', test_result_
         # evaluation path
         self.file_level_evaluation_file = f'{self.file_level_result_path}evaluation.csv'
         self.line_level_evaluation_file = f'{self.line_level_result_path}evaluation.csv'
+        self.time_file = f'{self.result_path}time.csv'
 
         # Model configuration info
         self.project_name = train_release.split('-')[0]

src/models/__pycache__/explain.cpython-38.pyc

@@ -19,7 +19,7 @@
 
 
 class LineDP(BaseModel):
-    model_name = 'LineDP'
+    model_name = 'MIT-LineDP'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -110,7 +110,7 @@ def line_level_prediction(self):
 
 #################################### Traditional model interpretation approaches #######################################
 class TMI_Model(BaseModel):
-    model_name = 'TMI-Model'
+    model_name = 'MIT-TMI-Model'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -208,7 +208,7 @@ def line_level_prediction(self):
 
 
 class TMI_LR(TMI_Model):
-    model_name = 'TMI-LR'
+    model_name = 'MIT-TMI-LR'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -218,7 +218,7 @@ def __init__(self, train_release: str = '', test_release: str = ''):
 
 
 class TMI_SVM(TMI_Model):
-    model_name = 'TMI-SVM'
+    model_name = 'MIT-TMI-SVM'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -228,7 +228,7 @@ def __init__(self, train_release: str = '', test_release: str = ''):
 
 
 class TMI_MNB(TMI_Model):
-    model_name = 'TMI-MNB'
+    model_name = 'MIT-TMI-MNB'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -238,7 +238,7 @@ def __init__(self, train_release: str = '', test_release: str = ''):
 
 
 class TMI_DT(TMI_Model):
-    model_name = 'TMI-DT'
+    model_name = 'MIT-TMI-DT'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -248,7 +248,7 @@ def __init__(self, train_release: str = '', test_release: str = ''):
 
 
 class TMI_RF(TMI_Model):
-    model_name = 'TMI-RF'
+    model_name = 'MIT-TMI-RF'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)

src/models/__pycache__/glance.cpython-38.pyc

@@ -115,7 +115,7 @@ class Glance_MD(Glance):
     """
     File level classifier: ManualDown
     """
-    model_name = 'Glance-MD'
+    model_name = 'BASE-Glance-MD'
 
     def __init__(self, train_release='', test_release='', line_threshold=0.5, file_threshold=0.5, test=False):
         test_result_path = ''
@@ -166,7 +166,7 @@ class Glance_EA(Glance):
     """
     File level classifier: Effort-Aware ManualDown
     """
-    model_name = 'Glance-EA'
+    model_name = 'BASE-Glance-EA'
 
     def __init__(self, train_release='', test_release='', line_threshold=0.5, file_threshold=0.5, test=False):
         test_result_path = ''
@@ -279,7 +279,7 @@ class Glance_LR(Glance):
     """
     File level classifier: Logistic Regression
     """
-    model_name = 'Glance-LR'
+    model_name = 'BASE-Glance-LR'
 
     def __init__(self, train_release='', test_release='', line_threshold=0.5, test=False):
         test_result_path = ''

src/models/__pycache__/natural.cpython-38.pyc

@@ -5,7 +5,7 @@
 
 
 class NGram(BaseModel):
-    model_name = 'NGram'
+    model_name = 'NLP-NGram'
 
     def __init__(self,train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -58,7 +58,7 @@ def line_level_prediction(self):
 
 
 class NGram_C(NGram):
-    model_name = 'NGram-C'
+    model_name = 'NLP-NGram-C'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)

src/models/__pycache__/tools.cpython-38.pyc

@@ -119,7 +119,7 @@ def line_level_prediction(self):
 # ################################################ PMD ##########################################################
 
 class PMD(StaticAnalysisTool):
-    model_name = 'PMD'
+    model_name = 'SAT-PMD'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)
@@ -162,7 +162,7 @@ def detect_detailed_result_of_bugs(self):
 
 # ################################################ CheckStyle ##########################################################
 class CheckStyle(StaticAnalysisTool):
-    model_name = 'CheckStyle'
+    model_name = 'SAT-CheckStyle'
 
     def __init__(self, train_release: str = '', test_release: str = ''):
         super().__init__(train_release, test_release)