
Инфраструктура и инструменты для MLOps | 321
# Пример: скетч рекомендательной системы, использующей ансамбль из двух моделей.
# Модель А будет запущена на локальной машине, модель В – на AWS.
class RecSysFlow(FlowSpec):
@step
def start(self):
self.data = load_data()
self.next(self.fitA, self.fitB)
# fitA требует другую версию NumPy, в отличие от fitB
@conda(libraries={"scikit-learn":"0.21.1", "numpy":"1.13.0"})
@step
def fitA(self):
self.model = fit(self.data, model="A")
self.next(self.ensemble)
@conda(libraries={"numpy":"0.9.8"})
# Требуются 2 GPU на 16 Гбайт памяти
@batch(gpu=2, memory=16000)
@step
def fitB(self):
self.model = fit(self.data, ...