Percorrer uma lista com o get_prediction

[Micael Santana Trevizan]

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).

[Micael Santana Trevizan]

Como faria isso?

[Denny Ceccon]

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

• This reply was modified 3 years ago by Denny Ceccon.

[Micael Santana Trevizan]

Boa noite!

Funcionol perfeitamente!  Muito obrigado!

[Jones Granatyr]

Qualquer outra dúvida é só avisar!

[Author]

Posts