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)