add z-1 dataset

run_interaction.py

@@ -37,6 +37,14 @@ def get_task(task_name):
     elif task_name.lower() == 'train_z_10':
         df = {"sps": './data/train_sps',
               "smile": './data/train_smile',
+              "affinity": './data/train_z_1_ic50',
+              }
+
+        return df, tokenizer_config
+
+    elif task_name.lower() == 'train_z_10':
+        df = {"sps": './data/train_sps',
+              "smile": './data/train_smile',
               "affinity": './data/train_z_10_ic50',
               }