In this guide, we'll deploy a Job to train a machine learning model. The model will learn to predict the species of an iris flower based on its sepal length, sepal width, petal length, and petal width. There are three species: Iris setosa, Iris versicolor, and Iris virginica.
We've already prepared the training script that trains a model on the Iris dataset, and you can find the code in our GitHub Repository.
Please visit the repository to familiarise yourself with the code you'll be deploying.
The project files are organised as follows:
. ├── train.py - Contains the training script code. └── requirements.txt - Contains the list of all dependencies.
All these files are located in the same directory.
Before you proceed with the guide, please ensure that you have setup a Workspace. To deploy your job, you'll need a workspace. If you don't have one, you can create it using this guide: Creating a Workspace or seek assistance from your cluster administrator.
Use these configs for the deployment form:
- Repo URL: https://github.com/truefoundry/getting-started-examples
- Path to build context: ./train-model/
- Command: python train.py
What we did above:
In the example above, we only had Python code and a requirements.txt file. The source code is in a Github repo and we don't have a prewritten docker file or a DockerImage - so we chose the Python Code option. Once we submit the form, TrueFoundry templatised a Dockerfile from the details provided and build the Docker image for us.
In the Build context field we specified the directory(
./train-model/) where our job code resides in the GitHub repository. We also specified the command(
python train.py) that we need to use to run our training script.
After you click
Submit, your job will be deployed in a few seconds. On successful deployment your Job will be displayed as Suspended (yellow) indicating that your Job has been deployed but it will not run automatically.
To run your Job you will have to trigger it manually.
Congratulations! You have successfully deployed and run your training Job.
To learn how to interact with your job check out this guide: Interacting with your Job
Updated 2 days ago