ynyou
/

CIFM

Mask Generation

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

Yuning You commited on Jan 24

Commit

14bd8e0

·

1 Parent(s): 6bd1fb8

update

Files changed (1) hide show

test.ipynb +9 -6

test.ipynb CHANGED Viewed

@@ -100,10 +100,14 @@
     }
    ],
    "source": [
-    "args_model = torch.load('./models_cifm/args.pt')\n",
-    "device = 'cpu' # or 'cuda' if you have a GPU\n",
-    "model = CIFM.from_pretrained('ynyou/CIFM', args=args_model).to(device)\n",
-    "model.eval()"
    ]
   },
   {
@@ -172,8 +176,7 @@
    ],
    "source": [
     "channel2ensembl_ids_target = [[i] for i in adata.var.index.tolist()]\n",
-    "channel2ensembl_ids_source = torch.load('./models_cifm/channel2ensembl.pt')\n",
-    "model.channel_matching(channel2ensembl_ids_target, channel2ensembl_ids_source)"
    ]
   },
   {

     }
    ],
    "source": [
+    "def load_model():\n",
+    "    args_model = torch.load('./models_cifm/args.pt')\n",
+    "    device = 'cpu' # or 'cuda' if you have a GPU\n",
+    "    model = CIFM.from_pretrained('ynyou/CIFM', args=args_model).to(device)\n",
+    "    model.channel2ensembl_ids_source = torch.load('./models_cifm/channel2ensembl.pt')\n",
+    "    model.eval()\n",
+    "    return model\n",
+    "model = load_model()"
    ]
   },
   {
    ],
    "source": [
     "channel2ensembl_ids_target = [[i] for i in adata.var.index.tolist()]\n",
+    "model.channel_matching(channel2ensembl_ids_target, model.channel2ensembl_ids_source)"
    ]
   },
   {