# In[ ]:
# saving model weights.
model2.save_weights('drive/My Drive/Case_Study/pfam/model2.h5')
# In[52]:
plot_history(history2)
# In[53]:
display_model_score(model2, [train_ohe, y_train], [val_ohe, y_val],
[test_ohe, y_test], 256)
# ## 5. Conclusion
# In[55]:
x = PrettyTable()
x.field_names = ['Sr.no', 'Model', 'Train Acc', 'Val Acc', 'Test Acc']
x.add_row(['1.', 'Bidirectional LSTM', '0.964', '0.957', '0.958'])
x.add_row(['2.', 'ProtCNN', '0.996', '0.988', '0.988'])
print(x)
# ### Reference:
#
# - https://www.biorxiv.org/content/10.1101/626507v4.full
