liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-04 11:30:57 +08:00
Code Activity

Set self.fitted = True instead of self._fitted.

Browse Source
This commit is contained in:
Qian Chen
2021-06-24 11:56:25 +08:00
committed by you-n-g
parent 8ebdb1e873
commit 2ab4a9adb3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_nn.py
View File

@@ -297,7 +297,7 @@ class DNNModelPytorch(Model):
_model_path = os.path.join(model_dir, _model_name)
# Load model
self.dnn_model.load_state_dict(torch.load(_model_path))
self._fitted = True
self.fitted = True
class AverageMeter: