航班時長模型：僅使用距離
在這個練習中，你要建立一個回歸模型來預測航班時長（duration 欄位）。
目前先保持模型簡單，只把航程距離（km 欄位）作為預測因子。
資料位於 flights。前幾筆紀錄已顯示在終端機中。這些資料也已分割為訓練集與測試集，可分別透過 flights_train 與 flights_test 取得。
本練習屬於課程
使用 PySpark 的機器學習
練習說明
- 建立一個線性回歸物件。指定標籤欄位名稱，並以訓練資料進行擬合。
- 對測試資料進行預測。
- 建立一個回歸評估器物件，並用它在測試資料上評估 RMSE。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
from pyspark.ml.regression import LinearRegression
from pyspark.ml.evaluation import RegressionEvaluator
# Create a regression object and train on training data
regression = ____(____).____(____)
# Create predictions for the testing data and take a look at the predictions
predictions = ____.____(____)
predictions.select('duration', 'prediction').show(5, False)
# Calculate the RMSE
____(____).____(predictions)