Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-06-01 21:56:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

(Enformer notebook) Make variant score list unique.

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enformer/enformer-usage.ipynb
+1 -1
View File
@@ -1142,7 +1142,7 @@
" break\n", " break\n",
" variant_scores = enformer_score_variants_all.predict_on_batch(\n", " variant_scores = enformer_score_variants_all.predict_on_batch(\n",
" {k: v[tf.newaxis] for k,v in example['inputs'].items()})[0]\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", " example_list.append({**example['metadata'],\n",
" **variant_scores})\n", " **variant_scores})\n",
" if i % 2 == 0:\n", " if i % 2 == 0:\n",