def get_WideDeep_net(config): """ Get network of wide&deep model. """ WideDeep_net = WideDeepModel(config) eval_net = PredictWithSigmoid(WideDeep_net) return eval_net
def get_WideDeep_net(config): WideDeep_net = WideDeepModel(config) loss_net = NetWithLossClass(WideDeep_net, config) train_net = TrainStepWrap(loss_net) eval_net = PredictWithSigmoid(WideDeep_net) return train_net, eval_net
def get_WideDeep_net(config): WideDeep_net = WideDeepModel(config) loss_net = NetWithLossClass(WideDeep_net, config) loss_net = VirtualDatasetCellTriple(loss_net) train_net = TrainStepWrap(loss_net) eval_net = PredictWithSigmoid(WideDeep_net) eval_net = VirtualDatasetCellTriple(eval_net) return train_net, eval_net
def get_WideDeep_net(config): """ Get network of wide&deep model. """ WideDeep_net = WideDeepModel(config) loss_net = NetWithLossClass(WideDeep_net, config) train_net = TrainStepWrap(loss_net, sparse=config.sparse) eval_net = PredictWithSigmoid(WideDeep_net) return train_net, eval_net
def get_WideDeep_net(config): """ Get network of wide&deep model. """ WideDeep_net = WideDeepModel(config) loss_net = NetWithLossClass(WideDeep_net, config) train_net = TrainStepWrap(loss_net, parameter_server=bool(config.parameter_server)) eval_net = PredictWithSigmoid(WideDeep_net) return train_net, eval_net
def get_wide_deep_net(config): """ Get network of wide&deep model. """ wide_deep_net = WideDeepModel(config) loss_net = NetWithLossClass(wide_deep_net, config) train_net = TrainStepWrap(loss_net, parameter_server=bool(config.parameter_server), cache_enable=(config.vocab_cache_size > 0)) eval_net = PredictWithSigmoid(wide_deep_net) return train_net, eval_net
def get_WideDeep_net(config): """ Get network of wide&deep model. """ WideDeep_net = WideDeepModel(config) loss_net = NetWithLossClass(WideDeep_net, config) loss_net = VirtualDatasetCellTriple(loss_net) train_net = TrainStepWrap(loss_net, host_device_mix=bool(config.host_device_mix)) eval_net = PredictWithSigmoid(WideDeep_net) eval_net = VirtualDatasetCellTriple(eval_net) return train_net, eval_net
def get_wide_deep_net(config): """ Get network of wide&deep model. """ wide_deep_net = WideDeepModel(config) loss_net = NetWithLossClass(wide_deep_net, config) if cache_enable: loss_net = VirtualDatasetCellTriple(loss_net) train_net = TrainStepWrap(loss_net, parameter_server=bool(config.parameter_server), sparse=config.sparse, cache_enable=(config.vocab_cache_size > 0)) eval_net = PredictWithSigmoid(wide_deep_net) if cache_enable: eval_net = VirtualDatasetCellTriple(eval_net) return train_net, eval_net