original_docstring: Return the config augmented with data-specific information.
docstring: Return the config augmented with data-specific information.
docstring_tokens: Return the config augmented with data - specific information .
code: def update_config(self, verbose=True):
          """Return the config augmented with data-specific information."""
          self.config['num_features'] = self.Z.shape[-1]
          self.config['num_cov'] = self.X.shape[-1]
          self.config['num_ts'] = self.Z.shape[0]
          if 'batch_size' not in self.config:
              self.config['batch_size'] = self.config['num_ts']
              if verbose:
                  print('Inferring num_features, num_cov, num_ts, batch_size from the data.')
          else:
              if verbose:
                  print('Inferring num_features, num_cov, num_ts from the data.')
          return self.config
code_tokens: def update_config ( self , verbose = True ) : self . config [ 'num_features' ] = self . Z . shape [ - 1 ] self . config [ 'num_cov' ] = self . X . shape [ - 1 ] self . config [ 'num_ts' ] = self . Z . shape [ 0 ] if 'batch_size' not in self . config : self . config [ 'batch_size' ] = self . config [ 'num_ts' ] if verbose : print ( 'Inferring num_features, num_cov, num_ts, batch_size from the data.' ) else : if verbose : print ( 'Inferring num_features, num_cov, num_ts from the data.' ) return self . config
short_docstring: Return the config augmented with data-specific information.
short_docstring_tokens: Return the config augmented with data - specific information .
comment: """Return the config augmented with data-specific information."""