rstudio / vetiver-python

Version, share, deploy, and monitor models.
https://rstudio.github.io/vetiver-python/stable/
MIT License
59 stars 17 forks source link

create scaffold polars prototype #214

Closed isabelizimm closed 1 month ago

isabelizimm commented 1 month ago

Notes:

import polars as pl
from vetiver import VetiverModel, VetiverAPI
from vetiver.data import mtcars
from sklearn.linear_model import LinearRegression

schema = {
    "mpg": pl.Float64,
    "cyl": pl.Int32,
    "disp": pl.Float32,
    "hp": int,
    "drat": float,
    "wt": float,
    "qsec": float,
    "vs": bool,
    "am": bool,
    "gear": pl.Int32,
    "carb": int,
}

mtcars = pl.DataFrame(mtcars, schema=schema)

model = LinearRegression().fit(mtcars.drop(columns="mpg"), mtcars["mpg"])
v = VetiverModel(model, model_name = "cars_linear", 
                 prototype_data = mtcars.drop(columns="mpg"))

VetiverAPI(v).run()
isabelizimm commented 1 month ago

closing, potentially looking at pandera at a later point to support this instead