diff options
author | Christoph Helma <helma@in-silico.ch> | 2016-10-27 11:58:07 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2016-10-27 11:58:07 +0200 |
commit | 8519274487166d75b3b9ae28e61f7a7be9f7e83c (patch) | |
tree | 6815047dae4740e57c33bca9f65b5716f32e0b0f /lib/train-test-validation.rb | |
parent | aada2ff67eaba251d1eeedb7f3eb29282706f997 (diff) |
probability plot for classification validations
Diffstat (limited to 'lib/train-test-validation.rb')
-rw-r--r-- | lib/train-test-validation.rb | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/lib/train-test-validation.rb b/lib/train-test-validation.rb index e3f5905..71abad2 100644 --- a/lib/train-test-validation.rb +++ b/lib/train-test-validation.rb @@ -44,10 +44,24 @@ module OpenTox class ClassificationTrainTest < TrainTest include ClassificationStatistics + field :accept_values, type: Array + field :confusion_matrix, type: Array + field :weighted_confusion_matrix, type: Array + field :accuracy, type: Float + field :weighted_accuracy, type: Float + field :true_rate, type: Hash + field :predictivity, type: Hash + field :probability_plot_id, type: BSON::ObjectId end class RegressionTrainTest < TrainTest include RegressionStatistics + field :rmse, type: Float, default:0 + field :mae, type: Float, default:0 + field :r_squared, type: Float + field :within_prediction_interval, type: Integer, default:0 + field :out_of_prediction_interval, type: Integer, default:0 + field :correlation_plot_id, type: BSON::ObjectId end end |