juaml/junifer
Watch
3
0
Fork
You've already forked junifer
 0
Code Issues 14 Pull requests 2 Packages Activity Actions

[FIX]: Make timeseries output 2D for RSSETSMarker #215

Merged
synchon merged 2 commits from fix/rss-ets-output into main 2023-03-30 12:50:49 +00:00
Conversation 3 Commits 2 Files changed 2 +3
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
docs/changes/newsfragments/215.bugfix Normal file
View file

@ -0,0 +1 @@
Fix the output of :class:`junifer.markers.RSSETSMarker` to be 2D by `Synchon Mandal`_

2
junifer/markers/ets_rss.py
View file

@ -133,6 +133,8 @@ class RSSETSMarker(BaseMarker):
edge_ts, _ = _ets(out["data"]) edge_ts, _ = _ets(out["data"])
# Compute the RSS # Compute the RSS
out["data"] = np.sum(edge_ts**2, 1) ** 0.5 out["data"] = np.sum(edge_ts**2, 1) ** 0.5
# Make it 2D
out["data"] = out["data"][:, np.newaxis]
# Set correct column label # Set correct column label
out["col_names"] = ["root_sum_of_squares_ets"] out["col_names"] = ["root_sum_of_squares_ets"]
return out return out