Solucionado (ver solução)
Solucionado
(ver solução)
2
respostas

Lista com dois colchetes

Ao criar a lista de entrada, utilizamos 2 colchetes dentro do código. Alguém pode me explicar o motivo desse formato? Qual é a interpretação do python com apenas um colchete?

2 respostas
solução!

Olá André,

Temos que fazer assim porque a função predict() requer um array 2d, ou simplificando, um array de arrays.

Como na aula estamos testando apenas uma entrada, utilizamos um array com apenas um array dentro: [[1, 2]], caso fossem duas entradas seria então um array com dois arrays dentro: [[1, 2], [2, 3]].

Isso é feito dessa maneira para possibilitar o uso do predict com várias entradas de uma só vez, exemplo:

entrada = [
    [np.log(Area), np.log(Dist_Praia + 1)],
    [np.log(Area), np.log(Dist_Praia + 1)],
    [np.log(Area), np.log(Dist_Praia + 1)]
]

# Retorna 3 previsões
modelo.predict(entrada)

É até por esse mesmo motivo que o retorno do predict é um array, e para acessar o valor da previsão o instrutor utiliza modelo.predict(entrada)[0] com o [0] no final.

Espero ter ajudado, qualquer dúvida é só falar.

Faz sentido.Muito obrigado!