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'''
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'))