/
githubmirror
/
Made-With-ML
Обзор
Документация
Войти
/
githubmirror
/
Made-With-ML
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
tests/model/utils.py
9 строк
274 B
GokuMohandas
updated to Ray 2.7
19 сен 2023, 08:03
19 сен 2023, 08:03
b98bd5b
Код
Авторство
О чём код?
import ray from madewithml import predict def get_label(text, predictor): sample_ds = ray.data.from_items([{"title": text, "description": "", "tag": "other"}]) results = predict.predict_proba(ds=sample_ds, predictor=predictor) return results[0]["prediction"]