я уж подумал по названию темы, что ты сам нейронки тренить начал

кстати, пользуясь случаем, если вдруг на форуме есть опытные дата-сатанисты, с опытом в настройке нейронок (датасеты, веса) яп не важен, отпишите в лс, есть вариант на большую зелень