add z-1 dataset_2

run_interaction.py

@@ -34,7 +34,7 @@ def get_task(task_name):
 
         return df_test, tokenizer_config
 
-    elif task_name.lower() == 'train_z_10':
+    elif task_name.lower() == 'train_z_1':
         df = {"sps": './data/train_sps',
               "smile": './data/train_smile',
               "affinity": './data/train_z_1_ic50',
@@ -183,7 +183,7 @@ if __name__ == '__main__':
                         help='number of data loading workers (default: 0)')
     parser.add_argument('--epochs', default=50, type=int, metavar='N',
                         help='number of total epochs to run')
-    parser.add_argument('--task', choices=['train', 'train_z_10', 'train_z_100', 'test', 'channel', 'ER', 'GPCR', 'kinase'],
+    parser.add_argument('--task', choices=['train', 'train_z_1', 'train_z_10', 'train_z_100', 'test', 'channel', 'ER', 'GPCR', 'kinase'],
                         default='train', type=str, metavar='TASK',
                         help='Task name. Could be train, test, channel, ER, GPCR, kinase.')
     parser.add_argument('--lr', '--learning-rate', default=1e-5, type=float,