def load_data(self):
    df = sns.load_dataset('flights')
    df['Date']  = pd.to_datetime(df['year'].astype(str) + '-' + df['month'].astype(str) + '-01')
    df = df.set_index('Date')
    df = df.drop(['year', 'month'], axis=1)
    df = df.values.astype(float)
    df = self.MinMaxScaler(df)
    train_size = int(len(df) * self.train_size)
    train, test = df[0:train_size,:], df[train_size:len(df),:]
    return train, test