mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 16:43:36 +00:00
chore: ml basic boilerplate
This commit is contained in:
17
experiments/ml/arch.py
Normal file
17
experiments/ml/arch.py
Normal 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
|
||||
Reference in New Issue
Block a user