import pickle, numpy as np
from comet.csk_feature_extract import CSKFeatureExtractor

extractor = CSKFeatureExtractor()

# ['iemocap', 'dailydialog', 'meld', 'emorynlp']
for dataset in ['epik']:
    print ('Extracting features in', dataset)
    sentences = pickle.load(open(dataset + '/' + dataset + '_sentences.pkl', 'rb'))        
    feaures = extractor.extract(sentences)
    pickle.dump(feaures, open(dataset + '/' + dataset + '_features_comet.pkl', 'wb'))
    
print ('Done!')