es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

  • es.davy.ai/
  • Programación /
  • Cómo pasar la entrada de texto al clasificador MultinomialNB después de guardar el modelo.

Cómo pasar la entrada de texto al clasificador MultinomialNB después de guardar el modelo.

29 Jul, 2023 Programación

Mi prueba de entrada es “SriRam”.
Esto debe ser clasificado usando un modelo de aprendizaje automático para determinar si es un nombre masculino o femenino.

Entonces, después de guardar el modelo:

Importar joblib
Importar numpy como np
Importar pandas como pd
Desde sklearn.feature_extraction.text importar CountVectorizer
Gender_model = open(‘Gender_model.pkl’,’rb’)
clf = joblib.load(Gender_model)

Entrada = “sriRam”

¿Cómo pasar esta entrada a Clf en el código anterior?

Gracias.

Tags:  ,

Answer

  1. Avatar for davy.ai
    davy.ai

    Para pasar la entrada “sriRam” a clf en el código anterior, debes seguir los siguientes pasos:

    1. Convertir la entrada en un arreglo usando numpy:
    test_input = np.array(['sriRam'])
    1. Usar CountVectorizer para transformar la entrada en un vector numérico:
    vectorizer = CountVectorizer()
test_input_vect = vectorizer.transform(test_input)
    1. Usar el modelo clf para predecir el género:
    output = clf.predict(test_input_vect)

    La salida será el género predicho de “sriRam”.

Comments are closed.