In [7]:
import pickle
import numpy as np
import os, sys
sys.path.append('../classification')

In [8]:
all_files = os.listdir('../classification')
estimator_files = [file for file in all_files if file.endswith('estimators.p')]

print(all_files)
print(estimator_files)

with open(estimator_files[0], 'rb') as f:
    estimators = pickle.load(estimator_file)


['sym2_1_coeffs.json', 'db2_1_results.json', 'BAGIDIS_Testing.Rmd', 'classify_nested.py', 'classify_manage.py', 'bandMap.py', 'plot_tsne.py', 'script_classify.py', 'classify.py', '__pycache__', 'featureExtraction.py', 'classify_SNe.py', 'tsne_db2_1_coeffs.pdf', 'getBagidisCoeffs.R', 'lightCurveAlign.py', 'db2_1_coeffs.json', 'feature_store.py', 'sym2_1_results.json']
[]
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-8-2690c556c156> in <module>()
      5 print(estimator_files)
      6 
----> 7 with open(estimator_files[0], 'rb') as f:
      8     estimators = pickle.load(estimator_file)

IndexError: list index out of range

In [ ]: