diff options
author | gebele <gebele@in-silico.ch> | 2017-01-16 09:50:33 +0000 |
---|---|---|
committer | gebele <gebele@in-silico.ch> | 2017-01-16 09:50:33 +0000 |
commit | 8266b2b447d8e138d5d5594f624a9f222e79f3ee (patch) | |
tree | f5c06159d0a529f263ea25f6354f89825fc41410 /lib/train-test-validation.rb | |
parent | 76912e72cf52c2a10423226ababb2b6d7f4b84ca (diff) | |
parent | d7504cc422bbaeee3546589d87e7baeb4e977c0b (diff) |
merge development
Diffstat (limited to 'lib/train-test-validation.rb')
-rw-r--r-- | lib/train-test-validation.rb | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/lib/train-test-validation.rb b/lib/train-test-validation.rb index 71abad2..034ae3a 100644 --- a/lib/train-test-validation.rb +++ b/lib/train-test-validation.rb @@ -2,11 +2,17 @@ module OpenTox module Validation + # Training test set validation class TrainTest < Validation field :training_dataset_id, type: BSON::ObjectId field :test_dataset_id, type: BSON::ObjectId + # Create a training test set validation + # @param [OpenTox::Model::Lazar] + # @param [OpenTox::Dataset] training dataset + # @param [OpenTox::Dataset] test dataset + # @return [OpenTox::Validation::TrainTest] def self.create model, training_set, test_set validation_model = model.class.create prediction_feature: model.prediction_feature, training_dataset: training_set, algorithms: model.algorithms @@ -32,16 +38,21 @@ module OpenTox validation end + # Get test dataset + # @return [OpenTox::Dataset] def test_dataset Dataset.find test_dataset_id end + # Get training dataset + # @return [OpenTox::Dataset] def training_dataset Dataset.find training_dataset_id end end + # Training test set validation for classification models class ClassificationTrainTest < TrainTest include ClassificationStatistics field :accept_values, type: Array @@ -54,6 +65,7 @@ module OpenTox field :probability_plot_id, type: BSON::ObjectId end + # Training test set validation for regression models class RegressionTrainTest < TrainTest include RegressionStatistics field :rmse, type: Float, default:0 |