epos-ai/platform-demo-scripts
3
0
Fork 0
Code Issues Pull Requests Activity

updated readme

Browse Source
This commit is contained in:
Krystyna Milian 2023-07-05 10:21:22 +02:00
parent 128d657e76
commit fd4a67f2ae
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@ -6,13 +6,16 @@ This repo contains notebooks and scripts demonstrating how to:
The original data can be downloaded from the [drive](https://drive.google.com/drive/folders/1InVI9DLaD7gdzraM2jMzeIrtiBSu-UIK?usp=drive_link)
- Explore available data, check the [notebook](notebooks/Explore%20igf%20data.ipynb)
- Train cnn model (Seisbench PhaseNet)to detect P phase, check the [script](scripts/train.py)
- Search for the best training hyperparams, check the [script](scripts/training_wandb_sweep.py)
- Train cnn model (Seisbench PhaseNet) to detect P phase, check the [script](scripts/train.py)
- Use Weights & Biases to search for the best training hyperparams, check the [script](scripts/training_wandb_sweep.py)
- Validate model performance, check the [notebook](notebooks/Check%20model%20performance%20depending%20on%20station-random%20window.ipynb)
- Use model for detecting P phase, check the [notebook](notebooks/Present%20model%20predictions.ipynb)
### Usage
To install all dependencies run:
1. Install all dependencies with poetry, run:
`poetry install`
2. Prepare .env file with content:
`WANDB_API_KEY="your key"`