Skip to content
Snippets Groups Projects
Commit 750207dc authored by Jörg Martin's avatar Jörg Martin
Browse files

First submitted version to NEPL

parent 62080bae
No related branches found
No related tags found
No related merge requests found
......@@ -12,6 +12,7 @@ import torch
import torch.backends.cudnn
from torch.utils.data import DataLoader
from matplotlib.pyplot import cm
import matplotlib
import matplotlib.pyplot as plt
from tqdm import tqdm
......@@ -20,6 +21,11 @@ from EIVTrainingRoutines import train_and_store
from EIVGeneral.coverage_collect import get_coverage_distribution
from EIVGeneral.manipulate_datasets import VerticalCut
font = {'family' : 'DejaVu Sans',
'weight' : 'normal',
'size' : 16}
matplotlib.rc('font', **font)
# coverages to consider
q_range = np.linspace(0.1, 0.95)
......@@ -89,9 +95,8 @@ def coverage_diagonal_plot(eiv_coverages, noneiv_coverages, color,
# create figures, together with title and axis labels
plt.figure(1)
plt.clf()
plt.title('Coverage for datasets with ground truth')
plt.xlabel('q')
plt.ylabel('coverage')
plt.ylabel('coverage ground truth')
# datasets to plot and their coloring
datasets = ['linear', 'quadratic','cubic','sine']
......@@ -251,5 +256,6 @@ if __name__ == '__main__':
plt.legend()
# save and show
plt.tight_layout()
plt.savefig('results/figures/true_coverage_vs_q.pdf')
plt.show()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment