Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bug in Categorical.py #110

Open
Sammi-Smith opened this issue Jan 23, 2020 · 0 comments
Open

Bug in Categorical.py #110

Sammi-Smith opened this issue Jan 23, 2020 · 0 comments

Comments

@Sammi-Smith
Copy link

In Categorical.output_inverse_transform(), I think there is a bug in line 104. It doesn't seem to be correctly outputting the original data categories. For me, it was just outputting all zeros. Upon closer inspection of that function and comparing it to the similar functions implemented for the other data types, it doesn't appear to be utilizing the response_transform_pipeline that is passed in as an argument to the function.

Currently, line 104 reads:
natural_scaled_vars = numpy.argmax(y_pred, axis=1)

I believe it should read:
natural_scaled_vars = response_variable_transformer.inverse_transform(y_pred)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant