Commit 75bfbaece29a7b96c4914b524619b5c808d37883
1 parent
f9d48b7f
cross validation bug fix
Showing
1 changed file
with
19 additions
and
2 deletions
sdk/plugins/validate.cpp
| @@ -13,9 +13,9 @@ class CrossValidateTransform : public MetaTransform | @@ -13,9 +13,9 @@ class CrossValidateTransform : public MetaTransform | ||
| 13 | { | 13 | { |
| 14 | Q_OBJECT | 14 | Q_OBJECT |
| 15 | Q_PROPERTY(QString description READ get_description WRITE set_description RESET reset_description STORED false) | 15 | Q_PROPERTY(QString description READ get_description WRITE set_description RESET reset_description STORED false) |
| 16 | - Q_PROPERTY(QList<br::Transform*> transforms READ get_transforms WRITE set_transforms RESET reset_transforms) | ||
| 17 | BR_PROPERTY(QString, description, "Identity") | 16 | BR_PROPERTY(QString, description, "Identity") |
| 18 | - BR_PROPERTY(QList<br::Transform*>, transforms, QList<br::Transform*>()) | 17 | + |
| 18 | + QList<br::Transform*> transforms; | ||
| 19 | 19 | ||
| 20 | void train(const TemplateList &data) | 20 | void train(const TemplateList &data) |
| 21 | { | 21 | { |
| @@ -50,6 +50,23 @@ class CrossValidateTransform : public MetaTransform | @@ -50,6 +50,23 @@ class CrossValidateTransform : public MetaTransform | ||
| 50 | { | 50 | { |
| 51 | transforms[src.file.getInt("Cross_Validation_Partition", 0)]->project(src, dst); | 51 | transforms[src.file.getInt("Cross_Validation_Partition", 0)]->project(src, dst); |
| 52 | } | 52 | } |
| 53 | + | ||
| 54 | + void store(QDataStream &stream) const | ||
| 55 | + { | ||
| 56 | + stream << transforms.size(); | ||
| 57 | + foreach (Transform *transform, transforms) | ||
| 58 | + transform->store(stream); | ||
| 59 | + } | ||
| 60 | + | ||
| 61 | + void load(QDataStream &stream) | ||
| 62 | + { | ||
| 63 | + int numTransforms; | ||
| 64 | + stream >> numTransforms; | ||
| 65 | + while (transforms.size() < numTransforms) | ||
| 66 | + transforms.append(make(description)); | ||
| 67 | + foreach (Transform *transform, transforms) | ||
| 68 | + transform->load(stream); | ||
| 69 | + } | ||
| 53 | }; | 70 | }; |
| 54 | 71 | ||
| 55 | BR_REGISTER(Transform, CrossValidateTransform) | 72 | BR_REGISTER(Transform, CrossValidateTransform) |