-
Notifications
You must be signed in to change notification settings - Fork 44
Description
It's not entirely clear to me what's going on in the cal_results function.
My classifications are of shape (n_samples, n_timesteps, n_classes), which is turned into a 1D array of shape (n_samples, ) by doing argmax on axis -1, which then passes all comparisons between y_true and y_pred.
But the logits are pickled as they are, which leads to a value error in cal_results, when calling model.fit().
I'm not sure I understand correctly. Should they be simply be unravelled, such that
[531, 21, 5, 681, ...] is compared to [1, 0, 0, 1]? Do I need to argmax the logits first? Because then the calibration fit spits out ValueError: X should be a 1d array.
But if I unravel it, softmax() throws a numpy.AxisError: axis 1 is out of bounds for array of dimension 1.
So which dimensionality is correct?