def main(args):
"""Main interface
"""
np.random.seed(12)
save_plot = args.save_plot
if save_plot is None:
save_plot = args.save_path
# Load and process data
# Provide objective to load
recall_0 = Utils.class_recall(0)
recall_1 = Utils.class_recall(1)
custom_obj = {'metr': recall_0}
path_to_models = os.listdir(args.save_path)
auc = []
for model_path in path_to_models:
if not os.path.exists(
(os.path.join(args.save_path, model_path, 'final_model.h5'))):
continue
if not os.path.exists(
os.path.join(args.save_path, model_path, 'predictions')):
os.makedirs(os.path.join(args.save_path, model_path,
'predictions'))
if not os.path.exists(
os.path.join(args.save_path, model_path, 'predictions',
args.data_path.split('/')[-1])):
os.makedirs(
os.path.join(args.save_path, model_path, 'predictions',
args.data_path.split('/')[-1]))
F = os.path.join(args.save_path, model_path,
'mean_std_final_model.pkl')
with open(F, 'rb') as mstd:
mean_tr, std_tr = pickle.load(mstd)
model = load_model(os.path.join(args.save_path, model_path,
'final_model.h5'),
custom_objects=custom_obj)
tech = args.technology
logging.info('Loading data...')
if args.is_synthetic == 1:
x, y, i2n = Utils.load_features(args.data_path,
max_len=args.max_len,
mode=args.mode,
technology=tech)
else:
x, y, i2n = Utils.load_features_nogt(args.data_path,
max_len=args.max_len,
mode=args.mode)
logging.info('Loaded {} contigs...'.format(len(set(i2n.values()))))
n2i = Utils.reverse_dict(i2n)
x = [xi for xmeta in x for xi in xmeta]
y = np.concatenate(y)
dataGen = Models.Generator(x,
y,
args.max_len,
batch_size=64,
shuffle=False,
norm_raw=bool(args.norm_raw),
mean_tr=mean_tr,
std_tr=std_tr)
loggin.info('Computing predictions for {}...'.format(tech))
scores = compute_predictions(y, n2i)
outfile = os.path.join(args.save_path, model_path, 'predictions',
args.data_path.split('/')[-1], tech + '.pkl')
with open(outfile, 'wb') as spred:
pickle.dump(scores, spred)
logging.info('File written: {}'.format(outfile))
def main(args):
"""Main interface
"""
# init
np.random.seed(args.seed)
## where to save the plot
save_plot = args.save_plot
if save_plot is None:
save_plot = args.save_path
# Load and process data
# Provide objective to load
logging.info('Loading data...')
recall_0 = Utils.class_recall(0)
recall_1 = Utils.class_recall(1)
custom_obj = {'metr': recall_0}
h5_file = os.path.join(args.model_path, args.model_name)
if not os.path.exists(h5_file):
msg = 'Cannot find {} file in {}'
raise IOError(msg.format(args.model_name, args.model_path))
logging.info('Loading model: {}'.format(h5_file))
model = load_model(h5_file, custom_objects=custom_obj)
# model pkl
pkl_file = os.path.join(args.model_path, args.mstd_name)
logging.info('Loading file: {}'.format(pkl_file))
with open(pkl_file, 'rb') as mstd:
mean_tr, std_tr = pickle.load(mstd)
# loading features
if args.is_synthetic == 1:
logging.info('Loading synthetic features')
x, y, i2n = Utils.load_features(args.feature_file_table,
max_len=args.max_len,
technology=args.technology,
force_overwrite=args.force_overwrite,
n_procs=args.n_procs)
else:
logging.info('Loading non-synthetic features')
x, y, i2n = Utils.load_features_nogt(
args.feature_file_table,
max_len=args.max_len,
force_overwrite=args.force_overwrite,
n_procs=args.n_procs)
logging.info('Loaded {} contigs'.format(len(set(i2n.values()))))
n2i = Utils.reverse_dict(i2n)
x = [xi for xmeta in x for xi in xmeta]
y = np.concatenate(y)
logging.info('Running model generator...')
dataGen = Models.Generator(x,
y,
args.max_len,
batch_size=64,
shuffle=False,
norm_raw=bool(args.norm_raw),
mean_tr=mean_tr,
std_tr=std_tr)
logging.info('Computing predictions for {}...'.format(args.technology))
scores = Utils.compute_predictions_y_known(y, n2i, model, dataGen)
outfile = os.path.join(
args.save_path, '_'.join([args.save_name, args.technology + '.pkl']))
with open(outfile, 'wb') as spred:
pickle.dump(scores, spred)
logging.info('File written: {}'.format(outfile))