Coverage for encodermap/encodermap_tf1/encodermap.py: 32%

1# Third Party Imports 

2import tensorflow.compat.v1 as tf 

3 

4# Local Folder Imports 

5from .autoencoder import Autoencoder 

6from .misc import distance_cost 

7 

8 

9class EncoderMap(Autoencoder): 

10 def _setup_cost(self): 

11 self._auto_cost() 

12 self._center_cost() 

13 self._l2_reg_cost() 

14 self._distance_cost() 

15 

16 def _distance_cost(self): 

17 if self.p.distance_cost_scale is not None: 

18 dist_cost = distance_cost( 

19 self.main_inputs, 

20 self.latent, 

21 *self.p.dist_sig_parameters, 

22 self.p.periodicity, 

23 ) 

24 tf.summary.scalar("sigmoid_loss", dist_cost) 

25 if self.p.distance_cost_scale != 0: 

26 self.cost += self.p.distance_cost_scale * dist_cost