Update dictionary for new model names

BiomedSciAI · Aug 28, 2024 · 3c1383e · 3c1383e
1 parent edbec4e
commit 3c1383e
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/notebooks/create_long_format_results.ipynb b/notebooks/create_long_format_results.ipynb
@@ -2,7 +2,7 @@
  "cells": [
  {
  "cell_type": "code",
- "execution_count": 35,
+ "execution_count": 10,
  "metadata": {},
  "outputs": [],
  "source": [
@@ -15,7 +15,7 @@
  },
  {
  "cell_type": "code",
- "execution_count": 36,
+ "execution_count": 11,
  "metadata": {},
  "outputs": [],
  "source": [
@@ -88,24 +88,30 @@
  " \"gene2vec\": \"Classical ML\",\n",
  " \"mistral\": \"LLM\",\n",
  " \"scGPT-whole-human\":\"ScRNA-seq\",\n",
- " \"scGPT-blood\":\"ScRNA-seq\"\n",
+ " \"scGPT-blood\":\"ScRNA-seq\",\n",
+ " \"esm\":\"Protein sequence\",\n",
+ " \"esm2\":\"Protein sequence\",\n",
+ " \"dnabert\":\"Base pair sequence\"\n",
  "}\n",
  "model_name = {\n",
  " \"mpnet\": \"MPNet\",\n",
  " \"bag_of_words\": \"Bag of Words\",\n",
- " \"top_mteb\": \"MTEB-1B\",\n",
+ " \"top_mteb\": \"MTEB-S\",\n",
  " \"geneformer\": \"Geneformer\",\n",
  " \"cellPLM\": \"cellPLM\",\n",
  " \"gene2vec\": \"Gene2vec\",\n",
- " \"mistral\": \"MTEB-7B\",\n",
+ " \"mistral\": \"MTEB-L\",\n",
  " \"scGPT-whole-human\":\"ScGPT-H\",\n",
- " \"scGPT-blood\":\"ScGPT-B\"\n",
+ " \"scGPT-blood\":\"ScGPT-B\",\n",
+ " \"esm\":\"ESM-1\",\n",
+ " \"esm2\":\"ESM-2\",\n",
+ " \"dnabert\":\"DNABert-2\",\n",
  "}"
  ]
  },
  {
  "cell_type": "code",
- "execution_count": 37,
+ "execution_count": 12,
  "metadata": {},
  "outputs": [],
  "source": [