Skip to content

Commit

Permalink
dummy regression
Browse files Browse the repository at this point in the history
  • Loading branch information
mike0sv committed Jan 23, 2025
1 parent 1cd1630 commit 4697189
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions src/evidently/future/metrics/regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
from evidently.future.tests import Reference
from evidently.future.tests import eq
from evidently.future.tests import gt
from evidently.future.tests import lt
from evidently.metrics import RegressionAbsPercentageErrorPlot
from evidently.metrics import RegressionDummyMetric
from evidently.metrics import RegressionErrorDistribution
Expand Down Expand Up @@ -140,6 +141,10 @@ class MAE(MeanStdMetric):
def _default_tests_with_reference(self, context: Context) -> List[BoundTest]:
return [eq(Reference(relative=0.1)).bind_single(self.get_fingerprint())]

def _default_tests(self, context: "Context") -> List[BoundTest]:
dv: SingleValue = context.calculate_metric(DummyMAE().to_calculation())
return [lt(dv.value).bind_single(self.get_fingerprint())]


class MAECalculation(LegacyRegressionMeanStdMetric[MAE]):
def calculate_value(
Expand All @@ -164,6 +169,10 @@ class RMSE(SingleValueMetric):
def _default_tests_with_reference(self, context: Context) -> List[BoundTest]:
return [eq(Reference(relative=0.1)).bind_single(self.get_fingerprint())]

def _default_tests(self, context: "Context") -> List[BoundTest]:
dv: SingleValue = context.calculate_metric(DummyRMSE().to_calculation())
return [lt(dv.value).bind_single(self.get_fingerprint())]


class RMSECalculation(LegacyRegressionSingleValueMetric[RMSE]):
def calculate_value(
Expand All @@ -185,6 +194,10 @@ class MAPE(MeanStdMetric):
def _default_tests_with_reference(self, context: Context) -> List[BoundTest]:
return [eq(Reference(relative=0.1)).bind_single(self.get_fingerprint())]

def _default_tests(self, context: "Context") -> List[BoundTest]:
dv: SingleValue = context.calculate_metric(DummyMAPE().to_calculation())
return [lt(dv.value).bind_single(self.get_fingerprint())]


class MAPECalculation(LegacyRegressionMeanStdMetric[MAPE]):
def calculate_value(
Expand Down

0 comments on commit 4697189

Please sign in to comment.