Percorrer uma lista com o get_prediction

Home Fóruns Fórum Processamento de Linguagem Natural com BERT e Python Percorrer uma lista com o get_prediction

Visualizando 4 posts - 1 até 4 (de 4 do total)
  • Autor
    Posts
  • #34227
    Boa tardde, nesta aula aprendemos como submeter uma frase ao treinamento

    Gostaria de saber como faço para submeter uma lista com varias frases, para não precisar analisar de uma em uma e printar o resultado ao lado, na forma de uma nova coluna com 0 e 1 (negativo e positivo).

    #34228

    Como faria isso?

    #34229
    Profile photo ofdennyDenny Ceccon
    Moderador

    Olá Micael,

    Você pode primeiro fazer uma pequena modificação na função get_predictions, para retornar o valor ao invés de exibi-lo na tela:

    def get_prediction(sentence):
      tokens = encode_sentence(sentence)
      inputs = tf.expand_dims(tokens, 0) # (batch_size) (1,...)
      output = Dcnn(inputs, training=False)
      sentiment = math.floor(output*2)
      if sentiment == 0:
        return 'negative'
      elif sentiment == 1:
        return 'positive'

    Agora, pode organizar suas frases em um dataframe e adicionar o resultado desta forma:

    texts = ["This movie was pretty interesting", "I'd rather not do that again", "What a great time to be alive", "I wish this never happened"]
    df = pd.DataFrame({'text': texts})
    df['emotion'] = df['text'].map(get_prediction)
    print(df)
    
    <out> 
      text                              emotion
    --------------------------------------------
    0 This movie was pretty interesting positive
    1 I'd rather not do that again      negative
    2 What a great time to be alive     positive
    3 I wish this never happened        negative
    
    
    • Esta resposta foi modificada 2 meses, 2 semanas atrás por Profile photo ofdennyDenny Ceccon.
    #34232
    Boa noite!

    Funcionol perfeitamente!  Muito obrigado!

Visualizando 4 posts - 1 até 4 (de 4 do total)
  • Você deve fazer login para responder a este tópico.