Skip to content

What should the final shapes be for calibration? #6

@komodovaran

Description

@komodovaran

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions