chore: ml basic boilerplate

This commit is contained in:
2025-12-12 12:29:39 +01:00
parent 584d996dac
commit 35a5225ae4
3 changed files with 49 additions and 0 deletions

17
experiments/ml/arch.py Normal file
View File

@@ -0,0 +1,17 @@
# this should retrun a model with exposed methods fit and transform method in an sklearn style
from sklearn.base import BaseEstimator, TransformerMixin
from procesing.context import PipelineContext
from typing import Any
TASK = 'classification'
LABELS = ['agent', 'human']
class BaseModel(BaseEstimator, TransformerMixin):
def __init__(self, context: PipelineContext):
self.context = context
def fit(self, X=None, y=None):
return self
def transform(self, X) -> Any:
pass