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
================================================================================
================================================================================
PERSISTENCE DIAGRAMS
================================================================================