開始使用免費開始

建立 TF-IDF 的 DataFrame

既然你已經產生了 TF-IDF 特徵,接下來需要把它們整理成可用來做推薦的格式。 你會再次使用 pandas,把陣列包成一個 DataFrame。 由於你會用電影標題來篩選資料,可以把標題指定為 DataFrame 的索引。

df_plots DataFrame 已經替你載入。它在 Title 欄包含電影名稱,在 Plot 欄包含劇情摘要。

本練習屬於課程

用 Python 建立推薦引擎

檢視課程

練習說明

  • 建立一個 TfidfVectorizer,並如同上一個練習那樣進行 fit 與 transform。
  • 把產生的 vectorized_data 包成一個 DataFrame。將在 fit 與 transform 階段產生的特徵名稱做為欄名,並把新的 DataFrame 指派給 tfidf_df。
  • 將原始的電影標題指定為新建立之 tfidf_df DataFrame 的索引。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

from sklearn.feature_extraction.text import TfidfVectorizer

# Instantiate the vectorizer object and transform the plot column
vectorizer = ____(max_df=0.7, min_df=2)
vectorized_data = vectorizer.____(df_plots['Plot']) 

# Create Dataframe from TF-IDFarray
tfidf_df = pd.____(____.toarray(), columns=vectorizer.____())

# Assign the movie titles to the index and inspect
tfidf_df.____ = ____['Title']
print(tfidf_df.head())
編輯並執行程式碼