p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

check wether input column exist in TFRecord files

This commit is contained in:
ms_yan 2020-05-19 20:27:23 +08:00
parent 017cdbe865
commit e6c25859d4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc
View File

@ -918,6 +918,9 @@ Status TFReaderOp::CreateSchema(const std::string tf_file, const std::vector<std
[](const auto &it) -> std::string { return it.first; });
for (const auto &curr_col_name : columns) {
auto it = feature_map.find(curr_col_name);
if (it == feature_map.end()) {
RETURN_STATUS_UNEXPECTED("Failed to find column " + curr_col_name);
}
std::string column_name = it->first;
std::string column_type;