stefan-insilico commited on
Commit
7d95793
·
verified ·
1 Parent(s): 69fed0d

Added option to process standardized configuration for any task

Browse files
Files changed (1) hide show
  1. 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
- 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
- prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
 
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