Я обучил модель TensorFlow классификации текста, но не могу использовать Polars DataFrame, чтобы делать на ее основе прогнозы. Однако я могу использовать DataFrame Pandas.
import pandas as pd
import polars as pl
import joblib
from tensorflow.keras.models import load_model
loaded_model =load_model('model.keras')
load_Le = joblib.load('label_encoder.joblib')
Если да:
text = "some example text"
df = pd.DataFrame({"Coment":[text]})
preddict = loaded_model.predict(df["Coment"])
У меня нет проблем, но если возникнут:
text = "some example text"
df = pl.DataFrame({"Coment":[text]})
preddict = loaded_model.predict(df["Coment"])
Я получаю TypeError: невозможно преобразовать аргумент type_value: String to a TensorFlow Dtype.
Какой совет?
Некоторая дополнительная информация:
Прежде чем сохранить мою модель, я добавил это, чтобы я мог прогнозировать любой текст (отлично работает на пандах)
inputs = keras.Input(shape=(1,), dtype="string")
processed_inputs = text_vectorization(inputs)
outputs = model(processed_inputs)
inference_model = keras.Model(inputs, outputs)
inference_model.save('model.keras')
Подробнее здесь: https://stackoverflow.com/questions/785 ... -dataframe
Модель TensorFlow не может прогнозировать данные поляры ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение