Выставляю исходный код, так как не могу продолжать разработку нейросети. Надеюсь кому-нибудь пригодится.