Added option to process standardized configuration for any task
Browse files- handler.py +8 -7
handler.py
CHANGED
@@ -42,10 +42,11 @@ class EndpointHandler:
|
|
42 |
if k=='instruction':
|
43 |
prompt+=f'<{v}>' if isinstance(v, str) else "".join([f'<{v_i}>' for v_i in v])
|
44 |
elif k=='up':
|
45 |
-
if v
|
46 |
-
|
47 |
elif k=='down':
|
48 |
-
|
|
|
49 |
else:
|
50 |
prompt+=f'<{k}>{v}</{k}>' if isinstance(v, str) else f'<{k}>{" ".join(v)} </{k}>'
|
51 |
return prompt
|
@@ -189,8 +190,8 @@ class EndpointHandler:
|
|
189 |
acc_embs_up2.append(self.emb_gpt_genes[self.emb_gpt_genes.gene_symbol==gs].embs.values[0])
|
190 |
except Exception as e:
|
191 |
pass
|
192 |
-
acc_embs_up1_mean = np.array(acc_embs_up1).mean(0)
|
193 |
-
acc_embs_up2_mean = np.array(acc_embs_up2).mean(0)
|
194 |
|
195 |
acc_embs_down1 = []
|
196 |
acc_embs_down2 = []
|
@@ -200,8 +201,8 @@ class EndpointHandler:
|
|
200 |
acc_embs_down2.append(self.emb_gpt_genes[self.emb_gpt_genes.gene_symbol==gs].embs.values[0])
|
201 |
except Exception as e:
|
202 |
pass
|
203 |
-
acc_embs_down1_mean = np.array(acc_embs_down1).mean(0)
|
204 |
-
acc_embs_down2_mean = np.array(acc_embs_down2).mean(0)
|
205 |
else:
|
206 |
acc_embs_up1_mean, acc_embs_up2_mean, acc_embs_down1_mean, acc_embs_down2_mean = None, None, None, None
|
207 |
|
|
|
42 |
if k=='instruction':
|
43 |
prompt+=f'<{v}>' if isinstance(v, str) else "".join([f'<{v_i}>' for v_i in v])
|
44 |
elif k=='up':
|
45 |
+
if v:
|
46 |
+
prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
|
47 |
elif k=='down':
|
48 |
+
if v:
|
49 |
+
prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
|
50 |
else:
|
51 |
prompt+=f'<{k}>{v}</{k}>' if isinstance(v, str) else f'<{k}>{" ".join(v)} </{k}>'
|
52 |
return prompt
|
|
|
190 |
acc_embs_up2.append(self.emb_gpt_genes[self.emb_gpt_genes.gene_symbol==gs].embs.values[0])
|
191 |
except Exception as e:
|
192 |
pass
|
193 |
+
acc_embs_up1_mean = np.array(acc_embs_up1).mean(0) if acc_embs_up1 else None
|
194 |
+
acc_embs_up2_mean = np.array(acc_embs_up2).mean(0) if acc_embs_up2 else None
|
195 |
|
196 |
acc_embs_down1 = []
|
197 |
acc_embs_down2 = []
|
|
|
201 |
acc_embs_down2.append(self.emb_gpt_genes[self.emb_gpt_genes.gene_symbol==gs].embs.values[0])
|
202 |
except Exception as e:
|
203 |
pass
|
204 |
+
acc_embs_down1_mean = np.array(acc_embs_down1).mean(0) if acc_embs_down1 else None
|
205 |
+
acc_embs_down2_mean = np.array(acc_embs_down2).mean(0) if acc_embs_down2 else None
|
206 |
else:
|
207 |
acc_embs_up1_mean, acc_embs_up2_mean, acc_embs_down1_mean, acc_embs_down2_mean = None, None, None, None
|
208 |
|