Escribir pruebas unitarias
En el video anterior sobre pruebas de inferencia, aprendiste la importancia de escribir casos de prueba para tu modelo entrenado y evaluado usando la biblioteca unittest de Python. En este ejercicio, pondrás en práctica lo aprendido escribiendo un caso de prueba para comprobar que el modelo genera salidas binarias como se espera. Tu model entrenado ya está importado, así como la parte de prueba del conjunto de datos X_test.
Este ejercicio forma parte del curso
Machine Learning de extremo a extremo
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
import unittest
import numpy as np
# Create a class called TestModelInference
class ____(unittest.____):
def setUp(self):
self.model = model