geneset_to_pathways

HanwenXuTHU · Nov 8, 2021 · 718c0d2 · 718c0d2
1 parent 40c7066
commit 718c0d2
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 116 deletions.
diff --git a/geneset_to_pathways/ProTranslator_kegg.py → geneset_to_pathways/ProTranslator_main.py b/geneset_to_pathways/ProTranslator_kegg.py → geneset_to_pathways/ProTranslator_main.py
@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from model import deepTNFSLModel
+from model import ProTranslatorModel
 from options import model_config, data_loading
 from file_loader import FileLoader
 from torch.utils.data import DataLoader
@@ -16,11 +16,11 @@
 import collections
 
 
-class DeepTNFSL:
+class ProTranslator:
 
     def __init__(self, model_config):
         self.loss_func = torch.nn.BCELoss()
-        self.model = deepTNFSLModel.deepTNFSLModel(input_nc=model_config.input_nc,
+        self.model = ProTranslatorModel.ProTranslatorModel(input_nc=model_config.input_nc,
                                                 in_nc=model_config.in_nc,
                                                 max_kernels=model_config.max_kernels,
                                                 hidden_dim=model_config.hidden_dim,

diff --git a/geneset_to_pathways/model/deepTNFSLModel.py → ...t_to_pathways/model/ProTranslatorModel.py b/geneset_to_pathways/model/deepTNFSLModel.py → ...t_to_pathways/model/ProTranslatorModel.py
@@ -12,7 +12,7 @@
 import torch
 
 
-class deepTNFSLModel(nn.Module):
+class ProTranslatorModel(nn.Module):
 
     def __init__(self,
                  input_nc=4,

diff --git a/geneset_to_pathways/model/deepGoPlusModel.py b/geneset_to_pathways/model/deepGoPlusModel.py
diff --git a/geneset_to_pathways/model/deepZSLModel.py b/geneset_to_pathways/model/deepZSLModel.py