koheiw / newsmap

Semi-supervised algorithm for geographical document classification
Other
59 stars 22 forks source link

Predicted values should be factor instead of character #21

Closed koheiw closed 3 years ago

koheiw commented 5 years ago

Predicted values should be factor with all the classes in the model, so that we can get countries with zero count in table() https://github.com/koheiw/newsmap/blob/861042060bb61bde2926350fac4f588a4e7a94df/R/textmodel_newsmap.R#L107-L112