17 lines
340 B
Python
17 lines
340 B
Python
from dataclasses import dataclass
|
|
from typing import List
|
|
|
|
from backend.metric.model import MetricValue
|
|
from backend.tree.model import Rule, TreeNode
|
|
|
|
|
|
@dataclass
|
|
class RegressionResult:
|
|
tree: List[TreeNode]
|
|
rules: List[Rule]
|
|
mse: MetricValue
|
|
mae: MetricValue
|
|
rmse: MetricValue
|
|
rmae: MetricValue
|
|
r2: MetricValue
|