Commit 75bfbaece29a7b96c4914b524619b5c808d37883

Authored by Josh Klontz
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)