def initialize(self, X, cctypes, distargs, num_states=1, col_names=None, init_mode='from prior'): """ Get or set the initial state """ vu.validate_data(X) vu.validate_cctypes(cctypes) self.X = X self.cctypes = cctypes self.num_states = num_states self.n_rows = len(X[0]) self.n_cols = len(X) if col_names is None: self.col_names = [ "col_" + str(i) for i in range(len(X)) ] else: assert( isinstance(col_names, list) ) assert( len(col_names) == len(X) ) self.col_names = col_names args = [(X, cctypes, distargs, init_mode) for _ in range(self.num_states)] if self.multithread: self.metadata = self.pool.map( _do_intialize, args ) else: self.metadata = [r for r in map( _do_intialize, args )]
def load_csv(self, filename, cctypes, distargs, num_states, init_mode='from prior'): X, col_names = gu.csv_to_data_and_colnames(filename) vu.validate_data(X) vu.validate_cctypes(cctypes) self.X = X self.cctypes = cctypes self.num_states = num_states self.col_names = col_names self.X = gu.clean_data(self.X, self.cctypes) args = [(self.X, self.cctypes, distargs, init_mode) for _ in range(self.num_states)] if self.multithread: self.metadata = self.pool.map( _do_intialize, args ) else: self.metadata = [r for r in map( _do_intialize, args )]