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 95a7a8d21e42a9ba926060067dad3f6022f7f1a0..983d219de6046fbcd87fe514197c070c671246ae 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_, {}); + auto acc_metrics_ = model_->Evaluate(test_ds_, {}); + auto res = acc_metrics_->Eval(); std::cout << "Accuracy is " << acc_metrics_->Eval() << 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 c2424bb6f506e1515181eb4baab0f5b7207a4344..299b9650fe2496fd08a79ed3eafcc3be463351b7 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; } ```