The easiest way to build and deploy machine learning microservices
UnionML is an open source MLOps framework that aims to reduce the boilerplate and friction that comes with building models and deploying them to production.
You can create UnionML Apps by defining a few core methods that are automatically bundled into ML microservices, starting with model training and offline and online prediction.
Built on top of Flyte, UnionML provides a high-level interface for productionizing your ML models so that you can focus on curating a better dataset and improving your models.
To learn more, check out the 📖 Documentation.
Install using conda:
conda install -c conda-forge unionmlInstall using pip:
pip install unionmlCreate a Dataset and Model, which together form a UnionML App:
from unionml import Dataset, Model
from sklearn.linear_model import LogisticRegression
dataset = Dataset(name="digits_dataset", test_size=0.2, shuffle=True, targets=["target"])
model = Model(name="digits_classifier", init=LogisticRegression, dataset=dataset)Define Dataset and Model methods for training a hand-written digits classifier:
from typing import List
import pandas as pd
from sklearn.datasets import load_digits
from sklearn.metrics import accuracy_score
@dataset.reader
def reader() -> pd.DataFrame:
return load_digits(as_frame=True).frame
@model.trainer
def trainer(
estimator: LogisticRegression,
features: pd.DataFrame,
target: pd.DataFrame,
) -> LogisticRegression:
return estimator.fit(features, target.squeeze())
@model.predictor
def predictor(
estimator: LogisticRegression,
features: pd.DataFrame
) -> List[float]:
return [float(x) for x in estimator.predict(features)]
@model.evaluator
def evaluator(
estimator: LogisticRegression,
features: pd.DataFrame,
target: pd.DataFrame
) -> float:
return float(accuracy_score(target.squeeze(), predictor(estimator, features)))And that's all ⭐️!
By defining these four methods, you've created a minimal UnionML App that you can:
- Execute locally to debug and iterate on your code.
- Serve Seamlessly with FastAPI for online prediction.
- Deploy on a Flyte Cluster to scale your model training and schedule offline prediction jobs.
All contributions are welcome 🤝 ! Check out the contribution guide to learn more about how to contribute.
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent