1
resposta

Como exporta os resultados do previsoes = modelo.predict(raw_teste_x)

Como exporta os resultados do previsoes = modelo.predict(raw_teste_x) para um arquivo csv parecido com o que usamos para ler, isso seria o output dos dados classificados

1 resposta

Olá Jose,

O retorno aqui é no formato de um array numpy, assim uma forma de salvar em csv é utilizando o próprio numpy ou transformando em um DataFrame antes:

# Utilizando o numpy
np.savetxt("prev_numpy.csv", previsoes, delimiter=",")

# Exemplo do arquivo prev_numpy.csv
0.000000000000000000e+00
1.000000000000000000e+00
1.000000000000000000e+00

# Exemplo de leitura do arquivo
previsoes = pd.read_csv('prev_numpy.csv', header=None)

# Utilizando o pandas
pd.DataFrame(previsoes, columns=['Previsões']).to_csv('prev_pandas.csv', index=False)

# Exemplo do arquivo prev_pandas.csv
Previsões
0
1
1

# Exemplo de leitura do arquivo
pd.read_csv('prev_pandas.csv')

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