Skip to content

Latest commit

 

History

History
21 lines (16 loc) · 2.27 KB

README.md

File metadata and controls

21 lines (16 loc) · 2.27 KB

Superhealth · GitHub GitHub repo size

Multiple health-related regression neural network models trained accessible securely on the internet.

Datasets

The datasets were gathered from UC Irvine Machine Learning Repository:

Python Setup

This project was developed in Python3 with a virtual environment through venv. You can find and install the dependencies by utilizing the requirements.txt provided.

Model Training and Creation

  • The models were created using PyTorch. There's one notebook for each model. These notebooks may be used to train or retrain the models as needed. A minimal setup was used when programming on these notebooks. Other than PyTorch, the only core libraries necessary are numpy and pandas. For model export and import, the corresponding onnx library or runtime is required.

Trained Models

Inside the /onnx/ directory, you will find the exported models in .onnx format. Please open their corresponding notebooks to see the model architecture, or open these .onnx files on Netron.

Web Application

The web application has its own /onnx/ directory which is a copy of the /onnx on the project root. The only requirement when hosting the web application is the ability to securely read resources. Therefore opening the web application's build files require a protocol other than the file protocol which is insecure. All model computations happen on the client-side (i.e. the device running the web application) for privacy and security. Please open the /web/ directory for more information.