learnware/examples/example_market_db/example_init.py

import os
import joblib
import numpy as np
from learnware.model import BaseModel


class SVM(BaseModel):
    def __init__(self):
        dir_path = os.path.dirname(os.path.abspath(__file__))
        self.model = joblib.load(os.path.join(dir_path, "svm.pkl"))

    def fit(self, X: np.ndarray, y: np.ndarray):
        pass

    def predict(self, X: np.ndarray) -> np.ndarray:
        return self.model.predict(X)

    def finetune(self, X: np.ndarray, y: np.ndarray):
        pass