def _init_shapes_and_data(self, data, labels):
self.n_features = data.shape[1]
if isinstance(data, pandas.core.frame.DataFrame):
self.feature_names = data.columns.tolist()
data = data.as_matrix()
if self.feature_names is None:
self.feature_names = ['feature_{}'.format(dim) for dim in range(self.n_features)]
if isinstance(labels, pandas.core.series.Series):
labels = labels.values
cntr = Counter(labels)
assert set(cntr.keys()) == {-1, 1}, "Labels must be encoded with -1, 1. Cannot contain more classes."
assert self.n_features is not None, "Number of attributes is None"
self.shapes = {name: ShapeFunction([np.PINF],
[0.0],
name)
for name in self.feature_names}
self.initialized = True
return data, labels
评论列表
文章目录