Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
arvind1609 committed Jan 25, 2025
1 parent 8e0f076 commit 7f16d5f
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 168 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ with torch.inference_mode():
print(f"Embedding dimensions : {embeddings.shape}")
```

An example notebook describing end-to-end feature extraction and downstream evaluation on the ppg-bp dataset is available [here](https://github.com/Nokia-Bell-Labs/papagei-foundation-model/blob/main/example_papagei.ipynb).
An example notebook describing end-to-end feature extraction and downstream evaluation on the ppg-bp dataset is available [here](https://github.com/Nokia-Bell-Labs/papagei-foundation-model/blob/main/example_papagei.ipynb). Some limitations of this work should be noted. First, there is not single model that is best across all tasks and datasets, thus we release the models with the most wins. Second, instead of fixed random seeds, we bootstrap the predictions 500 times to compute the 95% CI providing a performance range.

## Brief description of important modules

Expand Down
2 changes: 1 addition & 1 deletion example_papagei.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -633,7 +633,7 @@
"\n",
" estimator = Ridge()\n",
" param_grid = {\n",
" 'alpha': [0.1, 1.0, 10.0, 100.0], # Regularization strength\n",
" 'alpha': [0.1, 1.0, 10.0, 100.0, 1000.0], # Regularization strength\n",
" 'solver': ['auto', 'cholesky', 'sparse_cg'] # Solver to use in the computational routines\n",
" }\n",
" \n",
Expand Down
166 changes: 0 additions & 166 deletions training.py

This file was deleted.

0 comments on commit 7f16d5f

Please sign in to comment.