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

Fix python format by black

Browse Source
This commit is contained in:
D-X-Y
2021-03-08 06:39:03 +00:00
parent 4266492a34
commit 7bed3b4c2e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/contrib/model/pytorch_tabnet.py
View File

@@ -100,7 +100,9 @@ class TabnetModel(Model):
self.device self.device
) )
self.logger.info("model:\n{:}\n{:}".format(self.tabnet_model, self.tabnet_decoder)) self.logger.info("model:\n{:}\n{:}".format(self.tabnet_model, self.tabnet_decoder))
self.logger.info("model size: {:.4f} MB".format(count_parameters(self.tabnet_model) + count_parameters(self.tabnet_decoder))) self.logger.info(
"model size: {:.4f} MB".format(count_parameters(self.tabnet_model) + count_parameters(self.tabnet_decoder))
)
if optimizer.lower() == "adam": if optimizer.lower() == "adam":
self.pretrain_optimizer = optim.Adam( self.pretrain_optimizer = optim.Adam(