Skip to content

Multi-tail predictions extracted incorrect

Multi-tail predictions extracted incorrect

In class DataGenerator the method make_prediction() does not correctly extract the main-output path if more than one minor-outputs are used. There is a missing "-" in line 538:

keras_pred.values = np.swapaxes(np.expand_dims(tmp_keras[1,...], axis=1), 2, 0)

should be

keras_pred.values = np.swapaxes(np.expand_dims(tmp_keras[-1,...], axis=1), 2, 0)