|
|
@@ -33,8 +33,8 @@ class DataHandler(object):
|
|
|
train_y.extend(ty)
|
|
|
valid_y.extend(vy)
|
|
|
|
|
|
- train_y = np.array([[y.iloc[:, 1].values for y in train_y]])
|
|
|
- valid_y = np.array([[y.iloc[:, 1].values for y in valid_y]])
|
|
|
+ train_y = np.concatenate(np.array([[y.iloc[:, 1].values for y in train_y]]), axis=0)
|
|
|
+ valid_y = np.concatenate(np.array([[y.iloc[:, 1].values for y in valid_y]]), axis=0)
|
|
|
|
|
|
train_x = np.array([x.values for x in train_x])
|
|
|
valid_x = np.array([x.values for x in valid_x])
|
David