def test_mobilebert_model_without_training_for_tfjs(self):
model_spec = ms.mobilebert_classifier_spec(
seq_len=2, trainable=False, default_batch_size=1)
all_data = text_dataloader.TextClassifierDataLoader.from_folder(
self.text_dir, model_spec=model_spec)
self.train_data, self.test_data = all_data.split(0.5)
with self.assertRaises(Exception): # Raise an error when reloading model.
self._test_model_without_training(model_spec)
def test_mobilebert_model(self):
model_spec = ms.mobilebert_classifier_spec(seq_len=2, trainable=False)
all_data = text_dataloader.TextClassifierDataLoader.from_folder(
self.tiny_text_dir, model_spec=model_spec)
# Splits data, 50% data for training, 50% for testing
self.train_data, self.test_data = all_data.split(0.5)
model = text_classifier.create(self.train_data,
model_spec=model_spec,
epochs=1,
batch_size=1,
shuffle=True)
self._test_accuracy(model, 0.0)
self._test_export_to_tflite(model, threshold=0.0, atol=1e-2)
self._test_export_to_tflite_quant(model)
