def test_recent_compare(self): df = pd.read_csv(self.filename) ts = tslist(df) ts.fill_missed_median() timestamp = df['timestamp'].values start = 24 * 3600 * 14 / ts.span print start pred = [] for i in range(start, len(timestamp)): x_recent = ts.get_series(timestamp[i], w=30) if recent_compare(x_recent) > 0: pred.append(0) else: pred.append(1) df = df[start:] plot_ans(df, pred)
def test_absolute_periodicity_min(self): df = pd.read_csv(self.filename) ts = tslist(df) ts.fill_missed_median() timestamp = df['timestamp'].values start = 24 * 3600 * 14 / ts.span print start pred = [] for i in range(start, len(timestamp)): x_14 = [] for j in range(14): x_14.append(ts.get_value(timestamp[i] - j * 24 * 3600)) x_14.reverse() if absolute_periodicity_min(x_14) > 0: pred.append(0) else: pred.append(1) df = df[start:] plot_ans(df, pred)