From 3d5bc7f15fc1cedefcc77e917ecd03d84b4c6bdb Mon Sep 17 00:00:00 2001
From: Frozenmad <frozenmad2015@outlook.com>
Date: Fri, 17 Dec 2021 17:56:00 +0800
Subject: [PATCH] rename whole model using post-fix -model

---
 autogl/module/model/dgl/gcn.py       | 2 +-
 autogl/module/model/dgl/gin.py       | 2 +-
 autogl/module/model/dgl/graphsage.py | 2 +-
 autogl/module/model/dgl/topkpool.py  | 2 +-
 autogl/module/model/pyg/gat.py       | 2 +-
 autogl/module/model/pyg/gcn.py       | 2 +-
 autogl/module/model/pyg/gin.py       | 2 +-
 autogl/module/model/pyg/graphsage.py | 2 +-
 autogl/module/model/pyg/topkpool.py  | 2 +-
 9 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/autogl/module/model/dgl/gcn.py b/autogl/module/model/dgl/gcn.py
index ad7d30e..5b93489 100644
--- a/autogl/module/model/dgl/gcn.py
+++ b/autogl/module/model/dgl/gcn.py
@@ -114,7 +114,7 @@ class GCN(torch.nn.Module):
         return (prob_adj > 0).nonzero(as_tuple=False).t()
 
 
-@register_model("gcn")
+@register_model("gcn-model")
 class AutoGCN(BaseAutoModel):
     r"""
     AutoGCN.
diff --git a/autogl/module/model/dgl/gin.py b/autogl/module/model/dgl/gin.py
index 76cc7d2..bdcb806 100644
--- a/autogl/module/model/dgl/gin.py
+++ b/autogl/module/model/dgl/gin.py
@@ -243,7 +243,7 @@ class GIN(torch.nn.Module):
         # return score_over_layer
 
 
-@register_model("gin")
+@register_model("gin-model")
 class AutoGIN(BaseAutoModel):
     r"""
     AutoGIN. The model used in this automodel is GIN, i.e., the graph isomorphism network from the `"How Powerful are
diff --git a/autogl/module/model/dgl/graphsage.py b/autogl/module/model/dgl/graphsage.py
index 2040c2f..c2beca6 100644
--- a/autogl/module/model/dgl/graphsage.py
+++ b/autogl/module/model/dgl/graphsage.py
@@ -101,7 +101,7 @@ class GraphSAGE(torch.nn.Module):
 
 
 
-@register_model("sage")
+@register_model("sage-model")
 class AutoSAGE(BaseAutoModel):
     r"""
     AutoSAGE. The model used in this automodel is GraphSAGE, i.e., the GraphSAGE from the `"Inductive Representation Learning on
diff --git a/autogl/module/model/dgl/topkpool.py b/autogl/module/model/dgl/topkpool.py
index a03d697..8933f82 100644
--- a/autogl/module/model/dgl/topkpool.py
+++ b/autogl/module/model/dgl/topkpool.py
@@ -193,7 +193,7 @@ class Topkpool(torch.nn.Module):
         return score_over_layer
 
 
-@register_model("topkpool")
+@register_model("topkpool-model")
 class AutoTopkpool(BaseAutoModel):
     r"""
     AutoTopkpool. The model used in this automodel is from https://arxiv.org/abs/1905.05178, https://arxiv.org/abs/1905.02850
diff --git a/autogl/module/model/pyg/gat.py b/autogl/module/model/pyg/gat.py
index 76eb79c..277e587 100644
--- a/autogl/module/model/pyg/gat.py
+++ b/autogl/module/model/pyg/gat.py
@@ -114,7 +114,7 @@ class GAT(torch.nn.Module):
         return (prob_adj > 0).nonzero(as_tuple=False).t()
 
 
-@register_model("gat")
+@register_model("gat-model")
 class AutoGAT(BaseAutoModel):
     r"""
     AutoGAT. The model used in this automodel is GAT, i.e., the graph attentional network from the `"Graph Attention Networks"
diff --git a/autogl/module/model/pyg/gcn.py b/autogl/module/model/pyg/gcn.py
index 67e1b03..68e02db 100644
--- a/autogl/module/model/pyg/gcn.py
+++ b/autogl/module/model/pyg/gcn.py
@@ -284,7 +284,7 @@ class GCN(ClassificationSupportedSequentialModel):
         return (prob_adj > 0).nonzero(as_tuple=False).t()
 
 
-@register_model("gcn")
+@register_model("gcn-model")
 class AutoGCN(BaseAutoModel):
     r"""
     AutoGCN.
diff --git a/autogl/module/model/pyg/gin.py b/autogl/module/model/pyg/gin.py
index 14fd5ab..a9e816d 100644
--- a/autogl/module/model/pyg/gin.py
+++ b/autogl/module/model/pyg/gin.py
@@ -113,7 +113,7 @@ class GIN(torch.nn.Module):
         return F.log_softmax(x, dim=1)
 
 
-@register_model("gin")
+@register_model("gin-model")
 class AutoGIN(BaseAutoModel):
     r"""
     AutoGIN. The model used in this automodel is GIN, i.e., the graph isomorphism network from the `"How Powerful are
diff --git a/autogl/module/model/pyg/graphsage.py b/autogl/module/model/pyg/graphsage.py
index 3617e1d..c884481 100644
--- a/autogl/module/model/pyg/graphsage.py
+++ b/autogl/module/model/pyg/graphsage.py
@@ -202,7 +202,7 @@ class GraphSAGE(ClassificationSupportedSequentialModel):
         return (prob_adj > 0).nonzero(as_tuple=False).t()
 
 
-@register_model("sage")
+@register_model("sage-model")
 class AutoSAGE(BaseAutoModel):
     r"""
     AutoSAGE. The model used in this automodel is GraphSAGE, i.e., the GraphSAGE from the `"Inductive Representation Learning on
diff --git a/autogl/module/model/pyg/topkpool.py b/autogl/module/model/pyg/topkpool.py
index 73f72f8..6cc069c 100644
--- a/autogl/module/model/pyg/topkpool.py
+++ b/autogl/module/model/pyg/topkpool.py
@@ -84,7 +84,7 @@ class Topkpool(torch.nn.Module):
         return x
 
 
-@register_model("topkpool")
+@register_model("topkpool-model")
 class AutoTopkpool(BaseAutoModel):
     r"""
     AutoTopkpool. The model used in this automodel is from https://arxiv.org/abs/1905.05178, https://arxiv.org/abs/1905.02850