diff options
Diffstat (limited to 'lib/validation.rb')
-rw-r--r-- | lib/validation.rb | 19 |
1 files changed, 8 insertions, 11 deletions
diff --git a/lib/validation.rb b/lib/validation.rb index 651860e..9c19cde 100644 --- a/lib/validation.rb +++ b/lib/validation.rb @@ -62,6 +62,13 @@ module OpenTox validation end + end + + class ClassificationValidation < Validation + end + + class RegressionValidation < Validation + def statistics rmse = 0 weighted_rmse = 0 @@ -105,18 +112,8 @@ module OpenTox finished_at: Time.now ) =end - puts "R^2 #{r**2}" - puts "RMSE #{rmse}" - puts "MAE #{mae}" - return { "R^2" => r**2, "RMSE" => rmse, "MAE" => mae } + { "R^2" => r**2, "RMSE" => rmse, "MAE" => mae } end - - end - - class ClassificationValidation < Validation - end - - class RegressionValidation < Validation end end |