Skip to content
Using Python Flask, I wanted to create a simple web API that allows users to upload a dataset, choose one or more models, store them server side, and then hit an endpoint to get a prediction.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Project Goal

I want to build a simple to use API that allows users to upload a dataset, generate multiple machine learning models and receive an endpoint at which to make predictions in the future

Project Structure

  • - Flask API file
  • - Used for testing ML model stuff in console application
  • data_tools/
    • - tools for loading and parsing a CSV
    • - tools for saving and loading persistent ML models
  • datasets/
    • Keeps all datasets uploaded to API. Will be transitioning to some sort of blob storage
  • ml_models/
    • - Wrapper for any sklearn models we decide to support
    • - Wrapper for any custom ML models we want to implement
  • saved_models/
    • Used as storage of persistent ML models
You can’t perform that action at this time.