|
|
@@ -60,7 +60,7 @@ def model_training_bp():
|
|
|
feas = json.loads(bp.model_params)['features']
|
|
|
if set(feas).issubset(set(dh.opt.features)):
|
|
|
dh.opt.features = list(feas)
|
|
|
- train_x, train_y, valid_x, valid_y, scaled_train_bytes, scaled_target_bytes, scaled_cap = dh.train_data_handler(train_data)
|
|
|
+ train_x, train_y, valid_x, valid_y, scaled_train_bytes, scaled_target_bytes, scaled_cap = dh.train_data_handler(train_data, bp_data=True)
|
|
|
else:
|
|
|
model = bp.get_keras_model(bp.opt)
|
|
|
logger.info("训练数据特征，不满足，加强训练模型特征")
|
