Changeset 9131 for trunk/sources/HeuristicLab.Problems.Instances.DataAnalysis/3.3/Regression/RealWorld

Timestamp:

01/09/13 10:50:08 (12 years ago)

Author:

sforsten

Message:

#2001:

• renamed RegressionRealWorldInstanceProvider to RealWorldRegressionInstanceProvider
• added IRealWorldRegressionDataDescriptor and RealWorldResourceRegressionDataDescriptor to set the training and test partition for real world regression problem instances
• shuffled housing dataset and renamed TowerData.txt to Tower.txt in RegressionRealWorld.zip

Location:

trunk/sources/HeuristicLab.Problems.Instances.DataAnalysis/3.3/Regression/RealWorld

Files:

• IRealWorldRegressionDataDescriptor.cs (added)
• RealWorldRegressionInstanceProvider.cs (moved) (moved from trunk/sources/HeuristicLab.Problems.Instances.DataAnalysis/3.3/Regression/RealWorld/RegressionRealWorldInstanceProvider.cs) (2 diffs)
• RealWorldResourceRegressionDataDescriptor.cs (added)
• Tower.cs (added)

trunk/sources/HeuristicLab.Problems.Instances.DataAnalysis/3.3/Regression/RealWorld/RealWorldRegressionInstanceProvider.cs

@@ -21 +21 @@
 
 using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using HeuristicLab.Problems.DataAnalysis;
+using ICSharpCode.SharpZipLib.Zip;
 
 namespace HeuristicLab.Problems.Instances.DataAnalysis {
@@ -40 +45 @@
 
     protected override string FileName { get { return "RegressionRealWorld"; } }
+
+    public override IEnumerable<IDataDescriptor> GetDataDescriptors() {
+      List<IRealWorldRegressionDataDescriptor> descriptorList = new List<IRealWorldRegressionDataDescriptor>();
+      descriptorList.Add(new Tower());
+      var solutionsArchiveName = GetResourceName(FileName + @"\.zip");
+      if (!String.IsNullOrEmpty(solutionsArchiveName)) {
+        using (var solutionsZipFile = new ZipInputStream(GetType().Assembly.GetManifestResourceStream(solutionsArchiveName))) {
+          IList<string> entries = new List<string>();
+          ZipEntry curEntry;
+          while ((curEntry = solutionsZipFile.GetNextEntry()) != null) {
+            entries.Add(curEntry.Name);
+          }
+          foreach (var entry in entries.OrderBy(x => x)) {
+            string prettyName = Path.GetFileNameWithoutExtension(entry);
+            IRealWorldRegressionDataDescriptor desc = descriptorList.Where(x => x.Name.Equals(prettyName)).FirstOrDefault();
+            if (desc != null) {
+              yield return new RealWorldResourceRegressionDataDescriptor(prettyName, desc.Description, entry, desc.Training, desc.Test);
+            } else
+              yield return new ResourceRegressionDataDescriptor(prettyName, Description, entry);
+          }
+        }
+      }
+    }
+
+    public override IRegressionProblemData LoadData(IDataDescriptor id) {
+      var problem = base.LoadData(id);
+
+      var descriptor = id as RealWorldResourceRegressionDataDescriptor;
+
+      if (descriptor == null) {
+        return problem;
+      }
+
+      problem.TrainingPartition.Start = descriptor.Training.Start;
+      problem.TrainingPartition.End = descriptor.Training.End;
+      problem.TestPartition.Start = descriptor.Test.Start;
+      problem.TestPartition.End = descriptor.Test.End;
+
+      return problem;
+    }
   }
 }