Projeto simples de Machine Learning que prevê quanto de receita uma venda vai gerar.
Primeiro instale as bibliotecas:
pip install pandas matplotlib scikit-learnDepois rode o script:
python regressao_simples.pyBasicamente ele:
- Carrega um dataset com 50 mil vendas da Amazon
- Analisa os dados pra entender melhor
- Treina um modelo Random Forest
- Testa se o modelo consegue prever bem
- Mostra uns gráficos com os resultados
O modelo usa 3 informações pra fazer a previsão:
- Preço com desconto
- Quantidade vendida
- Preço original
O modelo acerta com um erro médio de apenas $0.09 por previsão, o que é bem bom!
As features mais importantes são:
- Preço com desconto (55%)
- Quantidade vendida (45%)
regressao_simples.py- código principalamazon_sales_dataset.csv- dados das vendasresultados.png- gráficos geradosREADME.md- este arquivo
Projeto feito pra aprender os conceitos básicos de ML como dividir dados em treino/teste, usar Random Forest e avaliar o modelo.