get predict_data.num_classes if trained_model havent num_classes #1199
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Я заметил, что если обучить CNN модель для классификации изображений, а после сохранить ее и повторно загрузить модель и запустить predict, то выпадает ошибка: AttributeError: 'Sequential' object has no attribute 'num_classes' (у trained_model нет поляnum_classes ) и нужно каждый раз заново обучать модель, а не использовать готовую
Изучив проблему, я понял, что мы можем брать num_classes не только из trained_model, а еще из predict_data. При этом все это нужно обернуть в try catch, чтобы мы взяли num_classes из predict_data только тогда, когда в trained_model нет поля num_classes
Код который использовался для тестирования https://github.com/aimclub/FEDOT/blob/master/examples/simple/classification/image_classification_problem.py
После добавленных изменений модель можно загрузить и сделать predict без каких-либо ошибок
Результаты тестов после добавления изменения остались такими же как и при их запуске после инициализации проекта