def plot_session_from_file(session_df_filepath): ''' Method to plot session from its session_df file @example from slm_lab.experiment import analysis filepath = 'data/reinforce_cartpole_2018_01_22_211751/reinforce_cartpole_t0_s0_session_df.csv' analysis.plot_session_from_file(filepath) ''' from slm_lab.experiment.monitor import InfoSpace spec_name = util.prepath_to_spec_name(session_df_filepath) session_spec = {'name': spec_name} session_df = util.read(session_df_filepath, header=[0, 1, 2, 3], index_col=0, dtype=np.float32) session_data = util.session_df_to_data(session_df) tn, sn = session_df_filepath.replace('_session_df.csv', '').split('_')[-2:] info_space = InfoSpace() info_space.set('experiment', 0) info_space.set('trial', int(tn[1:])) info_space.set('session', int(sn[1:])) session_fig = plot_session(session_spec, info_space, session_data) viz.save_image( session_fig, session_df_filepath.replace('_session_df.csv', '_session_graph.png'))
def session_data_from_file(predir, trial_index, session_index): '''Build session.session_data from file''' for filename in os.listdir(predir): if filename.endswith(f'_t{trial_index}_s{session_index}_session_df.csv'): filepath = f'{predir}/{filename}' session_df = util.read(filepath, header=[0, 1, 2, 3], index_col=0) session_data = util.session_df_to_data(session_df) return session_data
def session_data_from_file(predir, trial_index, session_index): '''Build session.session_data from file''' ckpt_str = '_ckpt-eval' if util.get_lab_mode() in ('enjoy', 'eval') else '' for filename in os.listdir(predir): if filename.endswith( f'_t{trial_index}_s{session_index}{ckpt_str}_session_df.csv'): filepath = f'{predir}/{filename}' session_df = util.read(filepath, header=[0, 1, 2, 3], index_col=0) session_data = util.session_df_to_data(session_df) return session_data
def session_data_from_file(predir, trial_index, session_index, ckpt=None, prefix=''): '''Build session.session_data from file''' ckpt_str = '' if ckpt is None else f'_ckpt-{ckpt}' for filename in os.listdir(predir): if filename.endswith( f'_t{trial_index}_s{session_index}{ckpt_str}_{prefix}session_df.csv' ): filepath = f'{predir}/{filename}' session_df = util.read(filepath, header=[0, 1, 2, 3], index_col=0) session_data = util.session_df_to_data(session_df) return session_data
def plot_session_from_file(session_df_filepath): ''' Method to plot session from its session_df file @example from slm_lab.experiment import analysis filepath = 'data/reinforce_cartpole_2018_01_22_211751/reinforce_cartpole_t0_s0_session_df.csv' analysis.plot_session_from_file(filepath) ''' spec_name = '_'.join(session_df_filepath.split('/')[1].split('_')[:-4]) session_spec = {'name': spec_name} session_df = util.read(session_df_filepath, header=[0, 1, 2, 3]) session_data = util.session_df_to_data(session_df) session_fig = plot_session(session_spec, session_data) viz.save_image( session_fig, session_df_filepath.replace('_session_df.csv', '_session_graph.png'))
def plot_session_from_file(session_df_filepath): ''' Method to plot session from its session_df file @example from slm_lab.experiment import analysis filepath = 'data/reinforce_cartpole_2018_01_22_211751/reinforce_cartpole_t0_s0_session_df.csv' analysis.plot_session_from_file(filepath) ''' from slm_lab.experiment.monitor import InfoSpace spec_name = util.prepath_to_spec_name(session_df_filepath) session_spec = {'name': spec_name} session_df = util.read(session_df_filepath, header=[0, 1, 2, 3], index_col=0, dtype=np.float32) session_data = util.session_df_to_data(session_df) tn, sn = session_df_filepath.replace('_session_df.csv', '').split('_')[-2:] info_space = InfoSpace() info_space.set('experiment', 0) info_space.set('trial', int(tn[1:])) info_space.set('session', int(sn[1:])) session_fig = plot_session(session_spec, info_space, session_data) viz.save_image(session_fig, session_df_filepath.replace('_session_df.csv', '_session_graph.png'))