Dory/plotPH_datasets.py at main · nihcompmed/Dory · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
import math
mpl.rcParams['text.usetex'] = True


def plotPD(data1, label1, data2, label2, fname):


        max1 = np.amax(data1)
        max2 = np.amax(data2)
        maxx = 1.1*max(max1, max2)

        data1[data1[:,1] == -1, 1] = maxx
        data2[data2[:,1] == -1, 1] = maxx

        plt.scatter(data1[:,0], data1[:,1], marker = '+', alpha=0.85, color='tab:blue', label=label1)
        plt.scatter(data2[:,0], data2[:,1], marker = 'o' , alpha=0.6, label = label2\
                                , facecolors='none', edgecolors='tab:red', linewidths=2)
        ax = plt.gca()
        #ax.set_yticks([maxx])
        #ax.set_yticklabels([r'$\infty$'], fontsize=20)
        plt.hlines(maxx, xmin=0, xmax=maxx, ls='--', alpha=0.3, color='black')
        plt.plot([0, maxx], [0, maxx], ls='--', alpha=0.5, color='black')

        plt.xlabel('birth', fontsize=24)
        plt.ylabel('death', fontsize=24)

        plt.tight_layout()
        plt.legend()

        plt.legend(prop={'size': 20})


        #plt.show()
        plt.savefig('figures/'+fname+'.png', format='png')
        plt.savefig('figures/'+fname+'.pdf', format='pdf')

        plt.cla()
        plt.clf()


dirr = 'Datasets/ripser_PD_results/'

datasets = ['Dragon', 'Fract', 'o3', 'torus4']

label1 = 'Ripser'
label2 = 'Dory'

data1 = np.loadtxt(dirr+'ripsdragon.txt', delimiter=',')
data2 = np.loadtxt('Datasets/Dragon/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripsdragonH1')


data1 = np.loadtxt(dirr+'ripsfract.txt', delimiter=',')
data2 = np.loadtxt('Datasets/Fract/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripsfractH1')


data1 = np.loadtxt(dirr+'ripsfractH2.txt', delimiter=',')
data2 = np.loadtxt('Datasets/Fract/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripsfractH2')


data1 = np.loadtxt(dirr+'ripso3.txt', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripso3H1')


data1 = np.loadtxt(dirr+'ripso3H2.txt', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripso3H2')


data1 = np.loadtxt(dirr+'ripstorus4.txt', delimiter=',')
data2 = np.loadtxt('Datasets/torus4/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripstorusH1')

data1 = np.loadtxt(dirr+'ripstorus4H2.txt', delimiter=',')
data2 = np.loadtxt('Datasets/torus4/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='ripstorusH2')

label1 = 'Gudhi'
label2 = 'Dory'

data1 = np.loadtxt('Datasets/o3/Gudhi_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='Gudhio3H1')


data1 = np.loadtxt('Datasets/o3/Gudhi_H2.csv', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='Gudhio3H2')


data1 = np.loadtxt('Datasets/torus4/Gudhi_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/torus4/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='GudhitorusH1')

data1 = np.loadtxt('Datasets/torus4/Gudhi_H2.csv', delimiter=',')
data2 = np.loadtxt('Datasets/torus4/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='GudhitorusH2')


label1 = 'Eirene'
label2 = 'Dory'

data1 = np.loadtxt('Datasets/Dragon/Eirene_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/Dragon/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='EirenedragonH1')

data1 = np.loadtxt('Datasets/Fract/Eirene_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/Fract/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='EirenefractH1')

data1 = np.loadtxt('Datasets/Fract/Eirene_H2.csv', delimiter=',')
data2 = np.loadtxt('Datasets/Fract/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='EirenefractH2')

data1 = np.loadtxt('Datasets/o3/Eirene_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='Eireneo3H1')


data1 = np.loadtxt('Datasets/o3/Eirene_H2.csv', delimiter=',')
data2 = np.loadtxt('Datasets/o3/DoryH2_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='Eireneo3H2')


data1 = np.loadtxt('Datasets/torus4/Eirene_H1.csv', delimiter=',')
data2 = np.loadtxt('Datasets/torus4/DoryH1_pers_data.txt', delimiter=',')
plotPD(data1, label1, data2, label2, fname='EirenetorusH1')