In [5]:
import pickle
from src import utils, bottleneck
# Load data from pickle file
with open('results/dragon_complexes.pkl', 'rb') as f:
complex_res = pickle.load(f)
complexes = complex_res['dragon 3500']
print(f"\n{'='*80}")
print("COMPLEX VISUALIZATION")
print(f"{'='*80}")
utils.visualize_complexes(complexes= complexes, title='dragon 3500', mode='plotly')
print(f"\n{'='*80}")
print("PERSISTENCE DIAGRAMS")
print(f"{'='*80}")
utils.plot_persistence_diagrams(complexes=complexes, mode='plotly')
================================================================================ COMPLEX VISUALIZATION ================================================================================