In [40]:
import pandas as pd
import matplotlib.pyplot as plt
In [41]:
domain = 'food'
df = pd.read_csv('~/' + domain + '.csv', skiprows=[1])
In [42]:
df.head()
Out[42]:
In [43]:
buffer = 10
mode = 'reattach'
In [44]:
df = df.loc[(df['buffer'] == buffer) & (df['mode'] == mode)]
In [45]:
df_both = df.loc[(df['parent'] == 'poincare') & (df['family'] == 'fasttext')]
df_both
Out[45]:
In [46]:
df_parent = df.loc[(df['parent'] == 'poincare') & (df['family'] == '-')]
df_parent
Out[46]:
In [47]:
df_family = df.loc[(df['parent'] == '-') & (df['family'] == 'fasttext')]
df_family
Out[47]:
In [50]:
# Buffer: | Mode:
metric = 'precision'
plt.plot(df_both['iterations'], df_both[metric], label='Both')
plt.plot(df_parent['iterations'], df_parent[metric], alpha=0.7, label='Only Parent')
plt.plot(df_family['iterations'], df_family[metric], alpha=0.7, label='Only Family')
plt.legend(loc='lower right')
plt.xlabel('Number of iterations')
plt.ylabel(metric)
plt.title('Buffer: ' + str(buffer) + ' | Mode: ' + mode + ' | ' + metric)
# plt.axis([0, 10, 0.364, 0.369])
plt.savefig('b5_mr.png')
plt.show()
In [ ]: