Bag of word scikit learn

import numpy as np
import pandas as pd
from sklearn.feature_extraction.text import CountVectorizer

docs = ['Tea is an aromatic beverage..',
        'After water, it is the most widely consumed drink in the world',
        'There are many different types of tea.',
        'Tea has a stimulating effect in humans.',
        'Tea originated in Southwest China during the Shang dynasty'] 

df = pd.DataFrame({'sms_message': docs, 'label': np.random.choice([0, 1], size=5)})

cv = CountVectorizer()
counts = cv.fit_transform(df['sms_message'])

df_counts = pd.DataFrame(counts.A, columns=cv.get_feature_names())
df_counts['label'] = df['label']

Source: stackoverflow.com

Similar codes

28 votes171 views2 answers
Wintp python manage.py createsuperuser
Python
-1 vote148 views1 answer
How to play mp3 file form pygame module
Python
2 votes187 views2 answers
Django request.data
Python Django
0 vote164 views1 answer
Django test postgres extensions intarray
Python
9 votes183 views2 answers
Django app
Python Django
0 vote215 views1 answer
Django action when create model
Python Django
10 votes174 views4 answers
File path current directory python
Python
0 vote166 views1 answer
Django.core.exceptions.ImproperlyConfigured: Field name is not valid for model
Python
42 votes172 views4 answers
Pickle.dump python
Python
15 votes167 views3 answers
Python current year
Python