Use word frequencies from the wordfreq library to allow for multilingual frequencies
No description
Add better handling of data with imbalanced class-labels
No description
Add legend to visualization
Add a legend somewhere in the visualization that explains what different colors of dots mean.