diff --git a/docs/lite/docs/source_en/train/runtime_train_cpp.md b/docs/lite/docs/source_en/train/runtime_train_cpp.md index 024c2a030844e23e4c7c8b1e554da779ed06f2b4..8ca6ca31f3db69b1c8128f8497a3c1f1ea01e3b3 100644 --- a/docs/lite/docs/source_en/train/runtime_train_cpp.md +++ b/docs/lite/docs/source_en/train/runtime_train_cpp.md @@ -65,7 +65,7 @@ int CreateSession() { std::cout << "Error " << status << " during build of model " << ms_file_; MS_ASSERT(status != mindspore::kSuccess); } - return; + return status; } ``` @@ -154,10 +154,11 @@ float Evaluate() { test_ds_ = test_ds_->Map({&typecast}, {"label"}); test_ds_ = test_ds_->Batch(batch_size_, true); - model_->Evaluate(test_ds_, {}); - std::cout << "Accuracy is " << acc_metrics_->Eval() << std::endl; + auto acc_metrics_ = model_->Evaluate(test_ds_, {}); + auto res = acc_metrics_->Eval(); + std::cout << "Accuracy is " << res << std::endl; - return 0.0; + return res; } ``` diff --git a/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md b/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md index 8c098f52df602c52288ff6df7ec8c241d581c936..d8143cbeb5862f01bd10e159f05923ae549f94ae 100644 --- a/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md +++ b/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md @@ -65,7 +65,7 @@ int CreateSession() { std::cout << "Error " << status << " during build of model " << ms_file_; MS_ASSERT(status != mindspore::kSuccess); } - return; + return status; } ``` @@ -154,10 +154,11 @@ float Evaluate() { test_ds_ = test_ds_->Map({&typecast}, {"label"}); test_ds_ = test_ds_->Batch(batch_size_, true); - model_->Evaluate(test_ds_, {}); - std::cout << "Accuracy is " << acc_metrics_->Eval() << std::endl; + auto acc_metrics_ = model_->Evaluate(test_ds_, {}); + auto res = acc_metrics_->Eval(); + std::cout << "Accuracy is " << res << std::endl; - return 0.0; + return res; } ```