(label)
| 54 | |
| 55 | # visualize the data |
| 56 | def visualize(label): |
| 57 | words = '' |
| 58 | for msg in df[df['labels'] == label]['data']: |
| 59 | msg = msg.lower() |
| 60 | words += msg + ' ' |
| 61 | wordcloud = WordCloud(width=600, height=400).generate(words) |
| 62 | plt.imshow(wordcloud) |
| 63 | plt.axis('off') |
| 64 | plt.title(label) |
| 65 | plt.show() |
| 66 | |
| 67 | visualize('spam') |
| 68 | visualize('ham') |