def test_testing(self, simple_model, loss, simple_data):
""" Tests that a testing run can succeed.
"""
simple_model.parse(None)
simple_model.build()
trainer = Executor(model=simple_model, loss=loss)
trainer.test(providers={'default': simple_data})
def test_uber_test(self, uber_model, uber_data, jinja_engine, loss):
""" Tests that we can compile and test a diverse model.
"""
uber_model.parse(jinja_engine)
uber_model.register_provider(uber_data)
uber_model.build()
trainer = Executor(model=uber_model, loss=loss)
trainer.compile()
trainer.test(providers={'default': uber_data})
def test_embedding_test(self, embedding_model, embedding_data, loss):
""" Tests that we can compile and test a model which has an Embedding.
function.
"""
embedding_model.parse(None)
embedding_model.register_provider(embedding_data)
embedding_model.build()
trainer = Executor(model=embedding_model, loss=loss)
trainer.test(providers={'default': embedding_data})
def test_ctc_test(self, ctc_model, ctc_data, ctc_loss):
""" Tests that we can compile and test a model using the CTC loss
function.
"""
ctc_model.parse(None)
ctc_model.register_provider(ctc_data)
ctc_model.build()
trainer = Executor(model=ctc_model, loss=ctc_loss)
trainer.test(providers={'default': ctc_data})
def test_testing(self, simple_model, loss, simple_data):
""" Tests that a testing run can succeed.
"""
simple_model.parse(None)
simple_model.build()
trainer = Executor(
model=simple_model,
loss=loss
)
trainer.test(providers={'default' : simple_data})
def test_embedding_test(self, embedding_model, embedding_data, loss):
""" Tests that we can compile and test a model which has an Embedding.
function.
"""
embedding_model.parse(None)
embedding_model.register_provider(embedding_data)
embedding_model.build()
trainer = Executor(
model=embedding_model,
loss=loss
)
trainer.test(providers={'default' : embedding_data})
def test_uber_test(self, uber_model, uber_data, jinja_engine, loss):
""" Tests that we can compile and test a diverse model.
"""
uber_model.parse(jinja_engine)
uber_model.register_provider(uber_data)
uber_model.build()
trainer = Executor(
model=uber_model,
loss=loss
)
trainer.compile()
trainer.test(providers={'default' : uber_data})
def test_ctc_test(self, ctc_model, ctc_data, ctc_loss):
""" Tests that we can compile and test a model using the CTC loss
function.
"""
ctc_model.parse(None)
ctc_model.register_provider(ctc_data)
ctc_model.build()
trainer = Executor(
model=ctc_model,
loss=ctc_loss
)
trainer.test(providers={'default' : ctc_data})