/Workforce_Predictor_KG

Machine Learning Model(s) for predicting future demand

Primary LanguagePython

Sales and staff demand forecasting

Project built with Pytorch that trains, evaluates, and deploys a Temporal Fusion Transformer (TFT) model. Deployed with SAM and called from an endpoint to return predicted values Part of the KG project

Models evaluated

  • Single layer regression nn
  • Multi layer regression nn
  • TFT