def __init__(self, feature, train_index, prep_index, prepped_data=None,
trained_data=None, inner_fitted_features=None):
self.feature = feature
# compute metadata
self.train_n = len(train_index)
self.prep_n = len(prep_index)
self.train_data_key = key_from_index(train_index)
self.prep_data_key = key_from_index(prep_index)
self.inner_fitted_features = inner_fitted_features
self.prepped_data = prepped_data
self.trained_data = trained_data
def __init__(self, fitted_model, data, data_description, result=None, reports=None):
super(PackagedModel, self).__init__()
self.fitted_model = fitted_model
self.n_rows = len(data)
self.n_cols = len(data.columns)
self.data_key = key_from_index(data.index)
self.data_description = data_description
self.result = result
self.reports = reports
def __init__(self,
feature,
train_index,
prep_index,
prepped_data=None,
trained_data=None,
inner_fitted_features=None):
self.feature = feature
# compute metadata
self.train_n = len(train_index)
self.prep_n = len(prep_index)
self.train_data_key = key_from_index(train_index)
self.prep_data_key = key_from_index(prep_index)
self.inner_fitted_features = inner_fitted_features
self.prepped_data = prepped_data
self.trained_data = trained_data
def __init__(self,
fitted_model,
data,
data_description=None,
built_at=None,
result=None,
reports=None):
super(PackagedModel, self).__init__()
self.fitted_model = fitted_model
self.n_rows = len(data)
self.n_cols = len(data.columns)
self.data_key = key_from_index(data.index)
self.data_description = data_description
self.built_at = built_at
self.result = result
self.reports = reports
