JamesRitchie/scikit-rvm

error with unequal sample sizes for classes

Closed this issue · 1 comments

It seems the RVC function only works when number of both classes are of the same sample size.
When train with a sample with 72 [0] and 78 [1], I receive the following error:
Error in python.exec("clf.fit(H,label)") :
operands could not be broadcast together with shapes (72,) (78,)

It worked when I set both classes with equal size 75 for each.

Fixed by #8.