Skip to content

Commit

Permalink
#1 add model weights
Browse files Browse the repository at this point in the history
  • Loading branch information
diegoluna3 committed Sep 30, 2024
1 parent d724568 commit 689ca72
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 1 deletion.
26 changes: 25 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -84,4 +84,28 @@ A custom script has been created to extract embeddings from a trained model. The
python -m deep_paint.utils.embeddings --config /path/to/your_config.yaml
~~~

This config file looks slightly different than the config file used for the four main commands. Refer to the [configs](./results/configs/) directory for examples.
This config file looks slightly different than the config file used for the four main commands. Refer to the [configs](./results/configs/) directory for examples.

## Results

### Overview

The [results](./results/) directory contains the following subdirectories:
- `checkpoints`: Contains model checkpoints
- `configs`: Contains configuration files used for training, getting model predictions, and getting model embeddings
- `embeddings`: Contains embeddings extracted from the model on the test set of the `RxRx2` data
- `logs`: Contains csv files extracted from `tensorboard` logs
- `metadata`: Contains custom metadata used for training the `DenseNet` model
- `predictions`: Contains model predictions on the test set of the `RxRx2` data

### Data Availability

The [`RxRx2`](https://www.rxrx.ai/rxrx2) dataset was used for training and evaluation of the `DenseNet` model. The dataset is freely available to download from the [`RxRx.ai`](https://www.rxrx.ai/) website.

### Model Weights

The [checkpoints](./results/checkpoints) directory contains model checkpoints for the binary and multiclass `DenseNet` model. These checkpoints can be used to load the trained models and make predictions.

### Notebooks

The [notebooks](./notebooks/) directory contains Jupyter notebooks that demonstrate the performance of the `DenseNet` model on the `RxRx2` dataset. The notebooks contain visualizations of the model predictions and embeddings.
3 changes: 3 additions & 0 deletions results/checkpoints/binary_rxrx2.ckpt
Git LFS file not shown
3 changes: 3 additions & 0 deletions results/checkpoints/multiclass_rxrx2.ckpt
Git LFS file not shown

0 comments on commit 689ca72

Please sign in to comment.