figure_2.py

# -*- coding: utf-8 -*-
"""
Created on Wed 1 Dic 2022
@author: davidsantiagoquevedo
@author: ntorresd
"""
import warnings
warnings.filterwarnings('ignore')

import sys
import yaml
import matplotlib.pyplot as plt

config = yaml.load(open("config.yml", "r"))["default"]

SCRIPTS_PATH = config['PATHS']['PLOT_PATH']
FIG_PATH = config['PATHS']['FIG_PATH'].format(dir = 'plots')
OUT_PATH = config['PATHS']['OUT_PATH'].format(dir = 'genomics')
sys.path.append(SCRIPTS_PATH)

import results_genomics as results_genomics

# Multinomial regression
fig, ax = plt.subplots(1,2, figsize = (13, 6))
axi = ax[0]
results_genomics.plot_multinomial(axi, ['2021-12', '2022-30'])
axi.set_title('a.')

# Heat map
axi = ax[1]
results_genomics.plot_heatmap(axi, n = 200)
axi.set_title('b.')
fig.tight_layout()
fig.savefig(FIG_PATH + 'figure_2.png')