1
$\begingroup$

experts of the statistics,

I am a newbie student in the machine learning field.

I just started a job to classify set of scientific abstracts into five classes.

The text distribution is as below:

Class1: 200

Class2: 950

Class3: 150

Class4: 100

class5: 350

I am planning to make a multi-class classifier,

however, I worry about the balance of the number of texts for each class.

For example, if I use 100 documents for training each class, the class 2 has relevantly too many testing data.

I wanted to get some insightful idea to construct my training/testing set and the reason.

Sincerely yours,

$\endgroup$
  • 1
    $\begingroup$ Compute the distribution of the classes, for example, c1 = 200/(200+950+150+100+350) ~ 0.11 , c2=950/(200+950+150+100+350) ~ 0.54 and so on. If you want to make a 70%/30% split then you select 0.11*0.3*200 of the elements belonging to class 1 randomly in order to go to the test set. The rest goes to the training set. Repeat for each class. In that way you get that the distribution of the classes in train and test are similar. $\endgroup$ – Fabian Werner May 8 '18 at 9:04
  • $\begingroup$ @FabianWerner Thank you for your attention :) Did you mean that the bias problem is relieved if the ratio of the whole classes is maintained in the training and testing set? $\endgroup$ – W Lee May 9 '18 at 6:40
  • $\begingroup$ @FabianWerner 0.11*0.3*200 = 6.6 , is it right? If the main point is to maintain the population distribution in each class, is it reasonable just divide each set by same partition ratio such as 0.7, 0.3? $\endgroup$ – W Lee May 9 '18 at 7:20
  • $\begingroup$ Depends on what you mean by 'bias problem'. If you mean "bias problem = classes are distributed unequal" then no, you will not solve this by sampling the training and test set correctly. You have to adapt your model in order to 'solve' this problem (misclassification of the rare class(es) is punished much more than misclassification of the overwhelming class(es)). What I suggested just makes it possible for you to see which model is capable of doing this and which is not. On your second comment: you are absolutely right and if you have millions of data points then what you say $\endgroup$ – Fabian Werner May 9 '18 at 10:49
  • $\begingroup$ works equally well. However, since you have only a few hundrets, randomness in the sampling might cause that the distribution of the classes in your test set is skewed... I also see that I was doing it in a wrong way: you just select 0.3*200 of the first class, 0.3*950 of the second class and so on and let these go into the test set, not 0.11*0.3*200, sorry for that. $\endgroup$ – Fabian Werner May 9 '18 at 10:51

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.