aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensor_forest
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-19 09:42:14 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-19 09:46:36 -0700
commit8d172e1a89feb06f906de43d75f0d5e65a2e1a04 (patch)
treebae824448fd7971200e7acc6af8dbb25ba8a2a67 /tensorflow/contrib/tensor_forest
parent98f522d3e982daafa9ccf136894cc83f496f5a11 (diff)
Refactor code to improve TensorDataSet construction speed.
PiperOrigin-RevId: 189595482
Diffstat (limited to 'tensorflow/contrib/tensor_forest')
-rw-r--r--tensorflow/contrib/tensor_forest/kernels/v4/input_data.h9
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h b/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h
index c544a8c75e..b991e6339f 100644
--- a/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h
+++ b/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h
@@ -44,12 +44,15 @@ class TensorDataSet {
int column_count = 0;
for (int i = 0; i < input_spec_.dense_size(); ++i) {
for (int j = 0; j < input_spec_.dense(i).size(); ++j) {
- decision_trees::FeatureId id;
- id.mutable_id()->set_value(strings::StrCat(column_count));
- available_features_.push_back(id);
++column_count;
}
}
+ available_features_.reserve(column_count);
+ decision_trees::FeatureId id;
+ for (int i = 0; i < column_count; i++) {
+ id.mutable_id()->set_value(strings::StrCat(i));
+ available_features_.emplace_back(id);
+ }
// Set up the random number generator.
if (split_sampling_random_seed_ == 0) {