Update training setting.

qlib/contrib/model/pytorch_gats.py

@@ -43,13 +43,13 @@ class GAT(Model):
         d_feat=6,
         hidden_size=64,
         num_layers=2,
-        dropout=0.7,
+        dropout=0.0,
         n_epochs=200,
-        lr=0.0001,
-        metric="loss",
+        lr=0.001,
+        metric="",
         early_stop=20,
         loss="mse",
-        base_model="LSTM",
+        base_model="GRU",
         with_pretrain=True,
         optimizer="adam",
         GPU="0",
@@ -148,17 +148,12 @@ class GAT(Model):
     def metric_fn(self, pred, label):
 
         mask = torch.isfinite(label)
-        if self.metric == "IC":
-            return self.cal_ic(pred[mask], label[mask])
 
         if self.metric == "" or self.metric == "loss":  # use loss
             return -self.loss_fn(pred[mask], label[mask])
 
         raise ValueError("unknown metric `%s`" % self.metric)
 
-    def cal_ic(self, pred, label):
-        return torch.mean(pred * label)
-
     def get_daily_inter(self, df, shuffle=False):
         # organize the train data into daily inter as daily batches
         daily_count = df.groupby(level=0).size().values