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