(Enformer notebook) Make variant score list unique.

PiperOrigin-RevId: 392387922
This commit is contained in:
Ziga Avsec
2021-08-23 10:44:42 +01:00
committed by Diego de Las Casas
parent 3d9f601c69
commit 27c2ca3956

View File

@@ -1142,7 +1142,7 @@
" break\n",
" variant_scores = enformer_score_variants_all.predict_on_batch(\n",
" {k: v[tf.newaxis] for k,v in example['inputs'].items()})[0]\n",
" variant_scores = {name[:20]: score for name, score in zip(df_targets.description, variant_scores)}\n",
" variant_scores = {f'{i}_{name[:20]}': score for i, (name, score) in enumerate(zip(df_targets.description, variant_scores))}\n",
" example_list.append({**example['metadata'],\n",
" **variant_scores})\n",
" if i % 2 == 0:\n",