Model Uploads

Free, zero-setup automation

Introduction

Model uploads are a simple and free way to automate your daily submissions. You upload a .pklfile with your trained model and we handle generating and submitting the predictions for you. You don't need to worry about reliability, infrastructure, scheduling, etc. you just focus on data science.

See this notebook for a full tutorial or this notebook for a barebones example.

How it works

1. Trained Model

Install requirements from the Numerai Predict execution environment and train a model. These are strict package version requirements that will help ensure your model always runs correctly. You may try to install later minor versions / patches of the packages listed as long as it is backwards-compatible with this environment.

Anything not listed is not available, but we do take requests over on Discord.

2. Prediction Pipeline

Wrap your model in a function that takes live features and outputs live predictions:

# Wrap your model with a function that takes live features and returns live predictions
def predict(
    live_features: pd.DataFrame,
    live_benchmark_models: pd.DataFrame
 ) -> pd.DataFrame:
    live_predictions = model.predict(live_features[feature_cols])
    submission = pd.Series(live_predictions, index=live_features.index)
    return submission.to_frame("prediction")

This is the only way for Numerai to properly run your model for you. If your pickle doesn't have a predict function with this signature, it will fail.

3. Cloudpickle

Pickle your function with the cloudpickle library and upload the pickle file to Numerai:

# Use the cloudpickle library to serialize your function
import cloudpickle
p = cloudpickle.dumps(predict)
with open("predict.pkl", "wb") as f:
f.write(p)

This library is used in other computing frameworks like Dask and Ray and allows you to easily run your local Python code remotely in the cloud. The main benefit of using cloudpickle over the standard pickle library is that it serializes your local context along with your code. This makes it very convenient to package up code developed locally in any environment.

In the example above, our predict function references model and feature_cols defined in the global scope. Cloudpickle is smart enough to correctly serialize both model and feature_cols by value so that it is also available when this function is run by Numerai.

4. Upload the Pickle

Upload your Model on the Submissions page (the Upload Model button on the right):

Select your pickle file, set the Python version you used, then click Upload

Numerai will execute your model to generate a live submission for the current round:

If this succeeds, a block will appear in the "submissions" column grid, you'll be able to review the execution logs for your model, and Numerai will generate diagnostics over the validation dataset for your model:

Once diagnostics complete, you'll also be able to view the diagnostics for this model:

5. Running Daily

Once uploaded, Numerai runs your model every day to generate and submit live predictions for you. If everything is working correctly, you will see the latest submission cycle through these 4 statuses:

1. Pending: Numerai is provisioning the cloud resources to run your model

2. Running: Numerai is now running your model

3. Validating: Numerai has ran your model and is now validating your submission

4. Success: Numerai has accepted your submission

If there was a problem, you will see 2 possible statuses:

1. Error: Numerai has encountered an unexpected error running your model and the team will look into it.

2. Failed: Your model failed to run. Please check the logs and re-upload a working model. Examples of model failures include

   • Python or dependency version mismatch

   • Invalid submission

   • Out of memory

   • Timeout

6. Disabling the Pickle

If, at any point, you would like to disable your uploaded model:

1. Visit the Submissions Page

2. Click the Model Upload button:

   
3. In the "Settings" tab, select "Disable"

   

FAQ

What versions of Python/packages work?

The default version is Python 3.11, but Python >=3.10 and <= 3.12 is supported and can be selected when uploading a model.

To see which package versions are used in the Model Uploads environment, see the Numerai Predict requirements.txt: https://github.com/numerai/numerai-predict/blob/master/requirements.txt

Can my model access the internet?

No, we do not give your model internet access.

What are the Mem / CPU limitations?

By default, we will provision each model a machine with 1 CPU with 4GB of ram and allow runtime of up to 10 minutes (does not include time spent queueing).

LightGBM Benchmarks:

• 20K trees using the small feature set runs in under 1 minute

• 90K trees using the full feature set runs in under 6 minutes

Can I still upload predictions manually?

No, once you upload your model, you will no longer be able to upload submissions via the API.

Can I have a webhook on my model?

No, to avoid race conditions after you upload your model, you cannot configure any webhooks on it. You will need to disable any existing compute configuration in order to upload your model.

Does Numerai have full access to my trained model?

Yes, Numerai has access to and can unpickle your trained model. If you are not comfortable with this, you should consider using another automation solution.

Can I re-download my pickled model?

Yes, on the submissions page, on the right, click the Cloud Icon > "Settings" > "Download"

Terms of service

Numerai reserves the right to disable your model for any reason, including (but not limited to) security, abuse, account inactivity, poor performance, etc.

Numerai will try our best to support your usage of this feature, but ultimately it is still your responsibility to make sure your submission pipeline is set up properly. We do not guarantee that your model will submit 100% of the time.

Numerai is not responsible for the gains or losses you might experience due to the performance of your uploaded model. You are solely responsible for the performance of the model.