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

Fix black error

Browse Source
This commit is contained in:
D-X-Y
2021-03-16 22:57:26 +08:00
parent 88b0871c12
commit 872ddc6f95
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
qlib/contrib/workflow/record_temp.py
View File

@@ -34,10 +34,7 @@ class SignalMseRecord(SignalRecord):
label = self.load("label.pkl")
masks = ~np.isnan(label.values)
mse = mean_squared_error(pred.values[masks], label[masks])
metrics = {
"MSE": mse,
"RMSE": np.sqrt(mse)
}
metrics = {"MSE": mse, "RMSE": np.sqrt(mse)}
objects = {"mse.pkl": mse, "rmse.pkl": np.sqrt(mse)}
self.recorder.log_metrics(**metrics)
self.recorder.save_objects(**objects, artifact_path=self.get_path())