Fix bug in DeepLearning's varimp that happens when max_category_features is in use

Description

Deeplearning with max_categorical_features lower that the number of categorical features fails on varimp.

This can be easily replicated by setting `max_categorical_features=10_000` and running the following dataset: https://www.openml.org/d/41147 .

Possible workaround is disabling the `variable_importances`.

Assignee

Tomas Fryda

Fix versions

Reporter

Tomas Fryda

Support ticket URL

None

Labels

None

Affected Spark version

None

Customer Request Type

None

Task progress

None

ReleaseNotesHidden

None

CustomerVisible

No

Priority

Major