tracebloc Python SDK

pip install "tracebloc[pytorch]>=0.18.1"                       # most users
# pip install "tracebloc[sklearn]>=0.18.1"                     # scikit-learn, incl. XGBoost / CatBoost / LightGBM
# pip install "tracebloc[xgboost]>=0.18.1"                     # one boosting backend only (or: catboost, lightgbm)
# pip install "tracebloc[lifelines,scikit-survival]>=0.18.1"   # survival analysis
# pip install "tracebloc[all]>=0.18.1"                         # pytorch + sklearn + lifelines + scikit-survival

from tracebloc import User

# 1. Log in to your tracebloc account
user = User()  # prompts for email and password

# 2. Upload a model file (must be in your current directory or provide a path)
user.upload_model(model_name="densenet.py")

# To upload with pretrained weights (weights file must be in the same directory):
# user.upload_model(model_name="densenet.py", weights=True)

# 3. Link model with a dataset from your use case
#    Find the Dataset ID on your use case page at ai.tracebloc.io
training_plan = user.link_model_dataset(dataset_id="YOUR_DATASET_ID")

# 4. Configure training parameters
training_plan.experiment_name("My first experiment")
training_plan.epochs(10)
training_plan.optimizer("adam")
training_plan.learning_rate({"type": "constant", "value": 0.001})
training_plan.validation_split(0.2)

# 5. Review your training plan
training_plan.get_training_plan()

# 6. Start training
training_plan.start()

# 7. Log out when done
user.logout()