1. Uczyć się
    2. /
  2. Courses
    3. /
  3. Preprocessing w uczeniu maszynowym w Pythonie
Connected

Exercise

Modelowanie zbioru danych UFO, część 2

Na koniec zbudujesz model z użyciem wektora tekstowego desc_tfidf i listy filtered_words do stworzenia przefiltrowanego wektora tekstowego. Sprawdź, czy uda ci się przewidzieć typ obserwacji (type) na podstawie tekstu. Skorzystasz w tym celu z modelu Naive Bayes.

Instrukcje

100 XP
  • Przefiltruj wektor desc_tfidf, przekazując listę filtered_words jako indeks.
  • Podziel cechy filtered_text i zmienną y na zbiory treningowy i testowy, zapewniając równomierny rozkład klas; użyj random_state równego 42.
  • Użyj metody .fit() modelu nb, aby dopasować go do X_train i y_train.
  • Wyświetl wartość .score() modelu nb na zbiorach X_test i y_test.