diff options
author | helma@in-silico.ch <helma@in-silico.ch> | 2018-09-06 17:24:25 +0200 |
---|---|---|
committer | helma@in-silico.ch <helma@in-silico.ch> | 2018-09-06 17:24:25 +0200 |
commit | 6d6be53a110e71d0d56ae5ea9a2675f76f7c84ec (patch) | |
tree | 14b31f061bec2fbb829fe84f675951ea8cda31b2 /lib/crossvalidation.rb | |
parent | 13e7865f386603fb784e62feef2ee2a56c015b45 (diff) |
adjusted classification similarities, dataset sdf export
Diffstat (limited to 'lib/crossvalidation.rb')
-rw-r--r-- | lib/crossvalidation.rb | 22 |
1 files changed, 12 insertions, 10 deletions
diff --git a/lib/crossvalidation.rb b/lib/crossvalidation.rb index e94864a..d1347a5 100644 --- a/lib/crossvalidation.rb +++ b/lib/crossvalidation.rb @@ -72,25 +72,27 @@ module OpenTox class ClassificationCrossValidation < CrossValidation 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 :confusion_matrix, type: Hash + field :weighted_confusion_matrix, type: Hash + field :accuracy, type: Hash + field :weighted_accuracy, type: Hash field :true_rate, type: Hash field :predictivity, type: Hash + field :nr_predictions, type: Hash field :probability_plot_id, type: BSON::ObjectId end # Crossvalidation of regression models class RegressionCrossValidation < CrossValidation 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 + field :rmse, type: Hash + field :mae, type: Hash + field :r_squared, type: Hash + field :within_prediction_interval, type: Hash + field :out_of_prediction_interval, type: Hash + field :nr_predictions, type: Hash field :warnings, type: Array + field :correlation_plot_id, type: BSON::ObjectId end # Independent repeated crossvalidations |