def on_finish(self, timestmap): self.on_trading_finish(timestmap) # show the result if self.draw_result: import plotly.io as pio pio.renderers.default = "browser" reader = AccountStatsReader(trader_names=[self.trader_name]) df = reader.data_df drawer = Drawer(main_data=NormalData(df.copy()[['trader_name', 'timestamp', 'all_value']], category_field='trader_name')) drawer.draw_line(show=True)
def draw_line(self, show=True): drawer = Drawer( main_data=NormalData( self.data_df.copy()[["trader_name", "timestamp", "all_value"]], category_field="trader_name" ) ) return drawer.draw_line(show=show)
def draw_result(self, entity_id): if self.label_method == "raw": df = self.kdata_df.loc[[entity_id], ["close"]].copy() pred_df = self.pred_y.to_frame(name="pred_close") pred_df = pred_df.loc[[entity_id], :].shift(self.predict_steps) drawer = Drawer( main_df=df, factor_df_list=[pred_df], ) drawer.draw_line(show=True) else: pred_df = self.pred_y.to_frame(name="pred_result").loc[[entity_id], :] df = self.testing_y.to_frame(name="real_result").loc[[entity_id], :].join(pred_df, how="outer") drawer = Drawer(main_df=df) drawer.draw_table()
def draw_line(self, show=True): drawer = Drawer(main_data=NormalData( self.data_df.copy()[['trader_name', 'timestamp', 'all_value']], category_field='trader_name')) return drawer.draw_line(show=show)
self.filters = filter super().__init__(Order, None, None, None, None, None, None, the_timestamp, start_timestamp, end_timestamp, columns, self.filters, order, None, level, category_field='trader_name', time_field='timestamp', computing_window=None) if __name__ == '__main__': reader = AccountStatsReader(trader_names=['000338_ma_trader']) drawer = Drawer(main_data=NormalData( reader.data_df.copy()[['trader_name', 'timestamp', 'all_value']], category_field='trader_name')) drawer.draw_line() # the __all__ is generated __all__ = ['AccountStatsReader', 'OrderReader']