撰寫單元測試
在前一支關於推論測試的影片中,你學到使用 Python 的 unittest 函式庫,為已訓練並評估完成的模型撰寫測試案例有多重要。在本練習中,你將動手實作:為模型撰寫一個測試案例,檢查它是否如預期產生二元輸出。已為你匯入訓練完成的 model,以及資料集中的測試部分 X_test。
本練習屬於課程
端到端機器學習
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
import unittest
import numpy as np
# Create a class called TestModelInference
class ____(unittest.____):
def setUp(self):
self.model = model