Using revscoring editquality enwiki damaging model as custom KFServing model.

kfserving-enwiki-damaging-model.py

import kfserving
from revscoring import Model
import mwapi
from revscoring.extractors import api
from typing import List, Dict


class EnWikiDamagingModel(kfserving.KFModel):
    def __init__(self, name: str):
        super().__init__(name)
        self.name = name
        self.ready = False

    def load(self):

        self.model = Model.load(
            open("../models/enwiki.damaging.gradient_boosting.model"), env_check=False)
        self.extractor = api.Extractor(mwapi.Session(
            "https://en.wikipedia.org", user_agent="Score edit demo in editquality"))
        self.ready = True

    def predict(self, request: Dict) -> Dict:

        inputs = request["rev_id"]
        feature_values = list(
            self.extractor.extract(rev_to_score, sm.features))
        results = sm.score(feature_values)
        return {"predictions": results}


if __name__ == "__main__":
    model = KFServingSampleModel("enwiki-damaging-model")
    model.load()
    kfserving.KFServer(workers=1).start([model])