Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

README.md

Machine Learning Model

Define the autoencoder model as a PyTorch model and transform it into a CoreML model using coremltools.

Requirements

  • install the requirements

shfmt

pip install -r requirements.txt

Generate the Model

Open the terminal and run:

shfmt

python generate_model.py

The CoreML model is per default saved to the required location, that is ../application/cardioflow/Ressources,