astrape 0.1.4

Astrape: A Fast Way to Organize and Deploy ML experiments.

ASTRAPE : A STrategic, Reproducible & Accessible Project and Experiment (Pre-release)

Creator : Woosog Benjamin Chay

[e-mail]

pre-release available pip install astrape

Table of Contents

• 1. Overview
• 2. Project
• 3. Experiment

1. Overview

---

Astrape : https://en.wikipedia.org/wiki/Astrape_and_Bronte

Astrape is a package that would help you organize machine learning projects. It is written mostly in PyTorch Lightning(https://pytorchlightning.ai).

---

Features of Astrape :

• Automatically creates appropriate folders and files(e.g., model checkpoints, logs, etc.) related to your experiment.
• All your experiments are logged to Tensorboard automatically.
• Enables you to define models easily.
• No more tedious magic commands.
• Can quickly apply simple basline algorithms in order to verify that your data is indeed "statistically significant" enough for machine learning tasks.

"Project" and "Experiment" conspire up to the soul of Astrape. The term "Project" here refers to "all set of possible machine learning experiments for analyzing the given data". Wait, what is an experiment anyways? An experiment here means "a process of train/validation/test phase with certain random state acquired for all random operations such as splitting scheme, initialization scheme, etc.". "Experiment" is a series of experiments with the same random state.

For stability's sake, you are tempted to (and should) conduct several "Experiments" with different random states to verify that your data analysis is indeed accurate. What Astrape does is it organizes such "Experiments" in a way that makes this sanity-checking process succint reproducible.

Pre-Release Notice "Project" is left unimplemented. Will be updated soon.

2. Project

Pre-Release Notice Will be implemented soon. Really soon.

Features of Project includes :

• Plotting the data
• Plotting results among experiments
• Providing arrays for axes in plotting
• More...

3. Experiment

When using Astrape, we expect you to conduct all experiments in the experiment.Experiment class. This class takes number of parameters, and you can check the details in the tutorial.

Once you declare an experiment, all random operations are governed with the same random seed you defined as a parameter for the experiment. When initialized (with given random state), train/validation/test data are specified, and you should now declare models for the task.

3-1. Specifying Models

Declare a model using .set_model() method. Astrape supports 1) multi-layer perceptron with all # of hidden units identical among layers (MLP), 2) multi-layer perceptron with # of hidden units contracting with given constant rate (ContractingMLP), 3) cutomized multi-layer perceptron of which you can define number of hidden units for each layer using list (CustomMLP), 4) VGG network (VGG), 5) UNet (UNet). The models mentioned in this paragraphs are all pytorch_lightning.LightningModules.

You can also declare sci-kit learn models as well using .set_model(). Astrape is compatible among sci-kit learn and pytorch-lightning modules.

3-2. (Optional) Specifying Trainers

PyTorch Lightning uses Trainer for training, validating, and testing models. You can specify it using .set_trainer() method with trainer configurations as parameters. If you don't, default values will be set for the Trainer.

3-3. Fitting the Model

You can fit the model using .fit() method. When you didn't specify a Trainer in previous step, default settings would be used in the fitting. Else, you can specify Trainer implicitly by passing the trainer configurations as parameters for .fit().

The training and valiation process is visualized in real-time using TensorBoard.

3-4. Stacking Fitted Models

Experiment class has .stack as an attribute. If .stack_models is set to True, fitted models will automatically be saved to .stack. If .stack_models is set to False, it would stop stacking fitted models to the stack. However, it would still save the model that is just fitted i.e., it will have memory of 1 fit. You can toggle .stack_models using .toggle_stack_models() method.

3-5. Saving Models

You can save the current model using .save_ckpt() method, or you can save the models in the stack using .save_stack() method. After .save_stack(), .stack will be flushed.

3-6. Checking the Best Model Thus Far

With .best_ckpt_thus_far() method, you can check the best model saved (in local) until now.

3-7. (Stratified) K-Fold Cross-Validation

You can perform (stratified) k-fold cross-validation using .cross_validation() method. See details in the tutorial.

.cross_validation() is compatible with sci-kit learn models as well. Astrape is compatible among sci-kit learn models and pytorch-lightning modules.