Searched refs:eval_callback (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 93 eval_callback = EvalCallBack(model, ds_eval, auc_metric, config)
102 … callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb])
106 assert np.allclose(eval_callback.eval_values, expect_out0)
108 assert np.allclose(eval_callback.eval_values, expect_out6)
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 95 eval_callback = EvalCallBack(model, ds_eval, auc_metric, config)
100 callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback])
101 eval_values = list(eval_callback.eval_values)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/ |
| D | test_deepfm.py | 67 eval_callback = EvalCallBack(model, ds_eval, auc_metric,
69 callback_list.append(eval_callback)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 371 if self.train_config.eval_callback:
374 self.train_config.eval_callback, model))
377 … "args eval_dataset is {}".format(self.train_config.eval_callback, eval_dataset))
379 eval_callback = EvalCallBack(model, eval_dataset, auc_metric,
382 callback_list.append(eval_callback)
|
| D | config.py | 61 eval_callback = True variable in TrainConfig
|