liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +08:00
Code Activity

gat2

Browse Source
This commit is contained in:
Hong Zhang
2020-11-26 13:55:12 +08:00
parent 398f67f8d8
commit f42661f2d4
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
qlib/contrib/model/pytorch_gats.py
View File

@@ -118,10 +118,6 @@ class GAT(Model):
) )
) )
if loss not in {"mse", "binary"}:
raise NotImplementedError("loss {} is not supported!".format(loss))
self._scorer = mean_squared_error if loss == "mse" else roc_auc_score
self.GAT_model = GATModel( self.GAT_model = GATModel(
d_feat=self.d_feat, d_feat=self.d_feat,
hidden_size=self.hidden_size, hidden_size=self.hidden_size,
@@ -211,7 +207,6 @@ class GAT(Model):
losses = [] losses = []
indices = np.arange(len(x_values)) indices = np.arange(len(x_values))
np.random.shuffle(indices)
for i in range(len(indices))[:: self.batch_size]: for i in range(len(indices))[:: self.batch_size]: