Esempi in Python per ExportFixedFeaturesNetwork

Esempio n. 1

File: network_units_test.py Progetto: vincentcheny/models

 def testExportFixedFeaturesNetworkWithDisabledEmbeddingMatrix(self):
     self._fixed_component.spec.fixed_feature[0].embedding_dim = -1
     network = network_units.ExportFixedFeaturesNetwork(
         self._fixed_component)
     self.assertEqual(0, len(network.params))

Esempio n. 2

File: network_units_test.py Progetto: vincentcheny/models

 def testExportFixedFeaturesNetworkWithEnabledEmbeddingMatrix(self):
     network = network_units.ExportFixedFeaturesNetwork(
         self._fixed_component)
     self.assertEqual(1, len(network.params))