编写单元测试
在上一段关于推理测试的视频中,您学习了使用 Python 的 unittest 库为已训练并评估过的模型编写测试用例的重要性。在本练习中,您将检验所学,编写一个模型的测试用例,用来检查其是否按预期产生二元输出。已为您导入训练好的 model,以及数据集的测试部分 X_test。
本练习是课程的一部分
端到端机器学习
交互式实操练习
通过完成这段示例代码来试试这个练习。
import unittest
import numpy as np
# Create a class called TestModelInference
class ____(unittest.____):
def setUp(self):
self.model = model