def runEpisode(episode,subtitles=True):
t0=time.time()
folderLoc = 'C:/Users/Brandon/Documents/Simpsons_Project/Character_Pickles/'
models = {}; imputers = {}; scalers = {}
for character in os.listdir(folderLoc):
if len(character.split('.p'))<=1: continue
print(character)
models[character] = pickle.load(open(folderLoc+character,'rb'))['model']
imputers[character] = pickle.load(open(folderLoc+character,'rb'))['imputer']
scalers[character] = pickle.load(open(folderLoc+character,'rb'))['scaler']
if episode.split('.')[-1] == 'p':
savedInfo = pickle.load(open(episode,'rb'))
episodeFeatures = savedInfo['features']
else:
episodeFeatures = getFeaturesParallel(episode,True) if subtitles: subtitles = getSubtitles(episode.split('.')[0]+'.srt')
preds = predict(subtitles,episodeFeatures,models,imputers,scalers)
print('Episode Runtime is: ',time.time()-t0,'(s)')
return(preds)
def calcAndSaveFeatures(episode):
episodeFeatures = getFeaturesParallel(episode)
pickle.dump({'features':episodeFeatures},open(episode.split('.wav')[0]+'.p','wb'))
