Skip to content

chukotskiyShaman/AI_tasks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Классификация:

Руководство по запуску:

Для запуска обучения нейронной сети необходимо запустить команду python3 train.py. Чтобы пропустить обучение и приступить к классификации на уже обученной модели нужно запустить команду python3 classify.py, для этого необходим файл сохраненной модели формата model.pth Для запуска файлов без использования cuda ядер необходимо стереть\закоментировать ".cuda()" везде где он есть.

Метрики:

Во время обучения выводятся точность и потери тренировочных и валидационных данных, а также precision для контроля переобучения модели.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published