Open Mr-Geekman opened 1 year ago
I can' reproduce an error, it works fine:
import numpy as np
import matplotlib.pyplot as plt
from etna.analysis import plot_time_series_with_change_points
from etna.datasets import TSDataset, generate_ar_df
def main():
df = generate_ar_df(periods=30, n_segments=3, start_time="2020-01-01", freq="D")
df.loc[df.index[5:7], "target"] = np.NaN
ts = TSDataset(df=df, freq="D")
change_points = {
"segment_0": [ts.index[10]],
"segment_1": [ts.index[17]],
"segment_2": [],
}
plot_time_series_with_change_points(ts=ts, change_points=change_points)
plt.savefig("plot_time_series_with_change_points.png", bbox_inches="tight")
if __name__ == "__main__":
main()
🚀 Feature Request
Fixes for EDA methods.
Proposal
Function
plot_time_series_with_change_points
does not work with NaNs in the middle of the series, try to fix it somehow or throw the appropriate exception.Test cases
TODO
Additional context
As a result of decomposition two more tasks were created:
286
287