Guest User

Untitled

a guest
Jan 18th, 2019
94
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.69 KB | None | 0 0
  1. dataset = pd.read_csv('Restaurant_Reviews.tsv', delimiter = 't', quoting = 3)
  2. for i in range(0, 1000):
  3. review = re.sub('[^a-zA-Z]', ' ', dataset['Review'][i])
  4. review = review.lower()
  5. review = review.split()
  6. ps = PorterStemmer()
  7. review = [ps.stem(word) for word in review if not word in set(stopwords.words('english'))]
  8. review = ' '.join(review)
  9. corpus.append(review)
  10. cv = CountVectorizer(max_features = 1500)
  11. X = cv.fit_transform(corpus).toarray()
  12. y = dataset.iloc[:, 1].values
  13. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.20, random_state = 0)
  14. classifier = GaussianNB()
  15. classifier.fit(X_train, y_train)
  16. y_pred = classifier.predict(X_test)
Add Comment
Please, Sign In to add comment