Commit c5689ea2 authored by Davis King's avatar Davis King

Fixed random forest feature extraction serialization.

parent 60a4af67
...@@ -83,7 +83,7 @@ namespace dlib ...@@ -83,7 +83,7 @@ namespace dlib
serialize(item.num_feats, out); serialize(item.num_feats, out);
} }
friend void serialize(dense_feature_extractor& item, std::istream& in) friend void deserialize(dense_feature_extractor& item, std::istream& in)
{ {
check_serialized_version("dense_feature_extractor", in); check_serialized_version("dense_feature_extractor", in);
deserialize(item.num_feats, in); deserialize(item.num_feats, in);
...@@ -210,6 +210,7 @@ namespace dlib ...@@ -210,6 +210,7 @@ namespace dlib
friend void serialize(const random_forest_regression_function& item, std::ostream& out) friend void serialize(const random_forest_regression_function& item, std::ostream& out)
{ {
serialize("random_forest_regression_function", out); serialize("random_forest_regression_function", out);
serialize(item.fe, out);
serialize(item.trees, out); serialize(item.trees, out);
serialize(item.leaves, out); serialize(item.leaves, out);
} }
...@@ -217,6 +218,7 @@ namespace dlib ...@@ -217,6 +218,7 @@ namespace dlib
friend void deserialize(random_forest_regression_function& item, std::istream& in) friend void deserialize(random_forest_regression_function& item, std::istream& in)
{ {
check_serialized_version("random_forest_regression_function", in); check_serialized_version("random_forest_regression_function", in);
deserialize(item.fe, in);
deserialize(item.trees, in); deserialize(item.trees, in);
deserialize(item.leaves, in); deserialize(item.leaves, in);
} }
......
...@@ -121,7 +121,7 @@ namespace dlib ...@@ -121,7 +121,7 @@ namespace dlib
}; };
void serialize(const dense_feature_extractor& item, std::ostream& out); void serialize(const dense_feature_extractor& item, std::ostream& out);
void serialize(dense_feature_extractor& item, std::istream& in); void deserialize(dense_feature_extractor& item, std::istream& in);
/*! /*!
provides serialization support provides serialization support
!*/ !*/
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment