Skip to main content

Tune The Model python wrapper

Tune The Model is a few-shot AutoML system.

Tune The Model can do almost anything that requires understanding or generating natural language. It is able to solve tasks in 12 languages: English, Spanish, Portuguese, Russian, Turkish, French, German, Italian, Arabic, Polish, Dutch, and Hebrew.

This package provides a simple wrapper for using our api.

Using tune-the-model package allows you to train and apply models.

Documentation

You can find the documentation at our Tune The Model API docs site.

Just try

We have fine-tuned several models. You can use the notebook to try them out. You can get the token to fine tune your own model.

Getting started

Firstly fill out the form to get a key to access the API. We will send you the key within a day.

Installation

To install the package just use pip install -U tune-the-model.

Usage

import tune_the_model as ttm
import pandas as pd

ttm.set_api_key('YOUR_API_KEY')

# load datasets
tdf = pd.read_csv('train.csv')
vdf = pd.read_csv('test.csv')

# Call one method. It will do everything for you:
# create a model, save it to the file, upload datasets and put the model in the queue for training.
model = ttm.tune_generator(
    'filename.json',
    tdf['inputs'], tdf['outputs'],
    vdf['inputs'], vdf['outputs']
)

# wait...
# a few hours
# while our GPUs train your model
model.wait_for_training_finish()

print(model.status)
print(model.is_ready)

# inference!
the_answer = model.generate('The Answer to the Ultimate Question of Life, the Universe, and Everything')
print(the_answer)

Metadata

Release files for tune-the-model 0.1.33

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for tune-the-model 0.1.33
File Size Uploaded
tune_the_model-0.1.33.tar.gz 14.8 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for tune-the-model 0.1.33
File Interpreter ABI Platform
tune_the_model-0.1.33-py3-none-any.whl Python 3 none any Details

Total release size: 30.3 kB

Release files / tune_the_model-0.1.33.tar.gz

Download URL tune_the_model-0.1.33.tar.gz
Size 14.8 kB
Tags Source
SHA-256 checksum
How to use checksums
cce80eeb6f998c686ecb7f7e9c12d333dbcbc075052ad5339af29cd22d9de805
BLAKE2b-256 checksum
How to use checksums
5fba52259d21fe2ee021ca557101ff267184ab49dc8a81bd3383c798948f4263
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.9.16

Release files / tune_the_model-0.1.33-py3-none-any.whl

Download URL tune_the_model-0.1.33-py3-none-any.whl
Size 15.4 kB
Tags Python 3
SHA-256 checksum
How to use checksums
4894ac4d97d77d10f7e30e11cb246205c980b54aea9ea79c5e451c2214b883cb
BLAKE2b-256 checksum
How to use checksums
81397b985ef5f75544359290768cdfd7b8df07d4f534844a516eb1512d128b5a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.9.16

Release history Release notifications | RSS feed

This release

0.1.33 This release

2 release files

0.1.32

2 release files

0.1.30

2 release files

0.1.29

2 release files

0.1.28

2 release files

0.1.27

2 release files

0.1.26

2 release files

0.1.25

2 release files

0.1.24

2 release files

0.1.23

2 release files

0.1.22

2 release files

0.1.21

2 release files

0.1.18

2 release files

0.1.17

2 release files

0.1.16

2 release files

0.1.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page