예제 #1
0
 def get_params(self):
     """Enumerate parameters for the run function.
     Returns:
       parameters
     """
     params = []
     for bottleneck in self.bottlenecks:
         for target_in_test, concepts_in_test in self.pairs_to_test:
             for alpha in self.alphas:
                 params.append(
                     run_params.RunParams(bottleneck, concepts_in_test,
                                          target_in_test,
                                          self.activation_generator,
                                          self.cav_dir, alpha,
                                          self.mymodel))
     return params
예제 #2
0
    def get_params(self):
        """
		Enumerate parameters for the run function.

		:return: parameters
		"""
        params = []
        for bottleneck in self.bottlenecks:
            for target_in_test, concept_in_test in self.pairs_to_test:
                for alpha in self.alphas:
                    tf.logging.info('%s %s %s %s', bottleneck, concept_in_test,
                                    target_in_test, alpha)
                    params.append(
                        run_params.RunParams(bottleneck, concept_in_test,
                                             target_in_test,
                                             self.activation_generator,
                                             self.cav_dir, alpha,
                                             self.black_box))
        return params