cross validation bug fix

Josh Klontz
1 parent f9d48b7f
Showing 1 changed file with 19 additions and 2 deletions
sdk/plugins/validate.cpp
@@ -13,9 +13,9 @@ class CrossValidateTransform : public MetaTransform
 {
     Q_OBJECT
     Q_PROPERTY(QString description READ get_description WRITE set_description RESET reset_description STORED false)
-    Q_PROPERTY(QList<br::Transform*> transforms READ get_transforms WRITE set_transforms RESET reset_transforms)
     BR_PROPERTY(QString, description, "Identity")
-    BR_PROPERTY(QList<br::Transform*>, transforms, QList<br::Transform*>())
+
+    QList<br::Transform*> transforms;
  
     void train(const TemplateList &data)
     {
@@ -50,6 +50,23 @@ class CrossValidateTransform : public MetaTransform
     {
         transforms[src.file.getInt("Cross_Validation_Partition", 0)]->project(src, dst);
     }
+
+    void store(QDataStream &stream) const
+    {
+        stream << transforms.size();
+        foreach (Transform *transform, transforms)
+            transform->store(stream);
+    }
+
+    void load(QDataStream &stream)
+    {
+        int numTransforms;
+        stream >> numTransforms;
+        while (transforms.size() < numTransforms)
+            transforms.append(make(description));
+        foreach (Transform *transform, transforms)
+            transform->load(stream);
+    }
 };
  
 BR_REGISTER(Transform, CrossValidateTransform)