Ошибка при трансляции модели keras в pyspark с пользовательским аргументом объекта - PullRequest
0 голосов
/ 24 апреля 2020

Я сталкиваюсь с проблемой при трансляции модели keras на pyspark, из-за этого я получаю ошибку 'tf' is not defined при вызове bc_classification_model.value. Есть какой-нибудь способ передать это?

import tensorflow as tf
from keras.models import load_model, Model
classification_model = load_model('/dbfs/FileStore/tables/temp/nemish/news_dedup_model_jacc.h5', custom_objects={'tf': tf})
bc_classification_model = sc.broadcast(classification_model)
...