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 )]
