def get_model_id(model_name=None):
df = db.get_data(table_name='models')
df = df[df['model_name'] == model_name]
if len(df) == 0:
raise ValueError('model name not found!')
else:
return int(df['id'].values[0])
def get_model_output_fields(model_name=None):
model_id = get_model_id(model_name)
df = db.get_data(table_name='model_output_config',
where_str='model_id = {0}'.format(model_id))
return df
def get_model_param_configs(model_name=None):
model_id = get_model_id(model_name)
df = db.get_data(table_name='model_param_config',
where_str='model_id ={0}'.format(model_id))
df['model_params'] = df['model_params'].astype(str)
return df
def get_strategy_signals(model_name=None):
model_id = get_model_id(model_name)
s = db.get_data(table_name='model_signals',
where_str='model_id = {0}'.format(model_id))
return s