Changeset 13921 for trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/RandomForest

Timestamp:

06/19/16 19:56:11 (8 years ago)

Author:

bburlacu

Message:

#2604: Revert changes to DataAnalysisSolution and IDataAnalysisSolution and implement the desired properties in model classes that implement IDataAnalysisModel, IRegressionModel and IClassificationModel.

File:

• trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/RandomForest/RandomForestModel.cs (modified) (1 diff)

trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/RandomForest/RandomForestModel.cs

@@ -45 +45 @@
     }
 
+    public IEnumerable<string> VariablesUsedForPrediction {
+      get { return originalTrainingData.AllowedInputVariables; }
+    }
+
+    public string TargetVariable {
+      get {
+        var regressionProblemData = originalTrainingData as IRegressionProblemData;
+        var classificationProblemData = originalTrainingData as IClassificationProblemData;
+        if (classificationProblemData != null)
+          return classificationProblemData.TargetVariable;
+        if (regressionProblemData != null)
+          return regressionProblemData.TargetVariable;
+        throw new InvalidOperationException("Getting the target variable requires either a regression or a classification problem data.");
+      }
+    }
+
     // instead of storing the data of the model itself
     // we instead only store data necessary to recalculate the same model lazily on demand