Traceback (most recent call last):


 "C:\ProgramData\Anaconda3\lib\site-packages\traitlets\traitlets.py", line 528, in get

value = obj._trait_values[self.name]
KeyError: 'context'

Traceback (most recent call last):


 "train_model.py", line 250, in <module>

train_model(model_name=model_name, dataset=dataset, mode=mode, weights=weights, computer=computer, run=run)


 "train_model.py", line 198, in train_model

train_generator, validation_generator = get_csv_generator(data, model_params, da, extended=False)


 "train_model.py", line 121, in get_csv_generator

batch_size=model_params['batch_size']


 "/home/michael/.local/lib/python3.5/site-packages/keras_preprocessing/image.py", line 1108, in flow_from_dataframe

interpolation=interpolation)


 "/home/michael/.local/lib/python3.5/site-packages/keras_preprocessing/image.py", line 2168, in __init__

self.classes = np.array([self.class_indices[cls] for cls in classes])


 "/home/michael/.local/lib/python3.5/site-packages/keras_preprocessing/image.py", line 2168, in <listcomp>

self.classes = np.array([self.class_indices[cls] for cls in classes])
KeyError: nan
