Доработка программного кода

Необходимо выполнить ряд заданий для прикрепленного кода:

Задание 1:

1. Проверить корректность представленного кода. Исправить неточности/ошибки (если таковые имеются).
2. Скорректировать представленный алгоритм машинного обучения таким образом, чтобы на выходе мы получали вероятность того, на сколько поданный на вход файл безопасен или опасен.

Задание 2.

1. Скорректировать вывод оценки алгоритма в форме, представленной во вложении на картинке.
2. Провести тестирование алгоритма на двух датасетах. И предоставить результаты тестирования.

Задание 3.

1. Исправить комментарии (при наличие неточностей в них, а также при добавлении своих изменений/внесении корректировок) в коде.

Все необходимые файлы для задания представлены в архиве и разложены по папкам.