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

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") label = self.load("label.pkl")
masks = ~np.isnan(label.values) masks = ~np.isnan(label.values)
mse = mean_squared_error(pred.values[masks], label[masks]) mse = mean_squared_error(pred.values[masks], label[masks])
metrics = { metrics = {"MSE": mse, "RMSE": np.sqrt(mse)}
"MSE": mse,
"RMSE": np.sqrt(mse)
}
objects = {"mse.pkl": mse, "rmse.pkl": np.sqrt(mse)} objects = {"mse.pkl": mse, "rmse.pkl": np.sqrt(mse)}
self.recorder.log_metrics(**metrics) self.recorder.log_metrics(**metrics)
self.recorder.save_objects(**objects, artifact_path=self.get_path()) self.recorder.save_objects(**objects, artifact_path=self.get_path())