開始使用免費開始

航班時長模型:Pipeline 階段

你將為航班時長模型的 pipeline 建立各個階段。你會在下一個練習中用這些階段來建構 pipeline,並建立一個迴歸模型。

StringIndexerOneHotEncoderVectorAssemblerLinearRegression 類別都已經匯入。

本練習屬於課程

使用 PySpark 的機器學習

檢視課程

練習說明

  • 建立一個 indexer,將 'org' 欄位轉換成名為 'org_idx' 的索引欄位。
  • 建立一個 one-hot 編碼器,將 'org_idx' 與 'dow' 欄位轉換為名為 'org_dummy' 與 'dow_dummy' 的虛擬變數欄位。
  • 建立一個 assembler,將 'km' 欄位與兩個虛擬變數欄位合併。輸出欄位名稱應為 'features'。
  • 建立一個線性迴歸物件來預測航班時長。

你可以回顧本課中的投影片,方法是開啟 IPython Shell 旁邊的 Slides 面板。

動手互動練習

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

# Convert categorical strings to index values
indexer = ____(____)

# One-hot encode index values
onehot = ____(
    inputCols=____,
    outputCols=____
)

# Assemble predictors into a single column
assembler = ____(inputCols=____, outputCol=____)

# A linear regression object
regression = ____(labelCol=____)
編輯並執行程式碼