Волшебное введение в алгоритмы классификации

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