changed seed and readme

README.md

@@ -13,7 +13,7 @@ short_description: Graph Isomorphism Network Baseline Classifier for Tox21
 
 This repository hosts a Hugging Face Space that provides an examplary API for submitting models to the [Tox21 Leaderboard](https://huggingface.co/spaces/tschouis/tox21_leaderboard).
 
-In this example, we trained a GIN classifier on the Tox21 targets and saved the trained model in the `assets/` folder.
+In this example, we trained a GIN classifier on the Tox21 targets and saved the trained model in the `checkpoints/` folder.
 
 **Important:** For leaderboard submission, your Space needs to include training code. The file `train.py` should train the model using the config specified inside the `config/` folder and save the final model parameters into a file inside the `checkpoints/` folder. The model should be trained using the [Tox21_dataset](https://huggingface.co/datasets/tschouis/tox21) provided on Hugging Face. The datasets can be loaded like this:
 ```python
@@ -29,7 +29,6 @@ val_df = ds["validation"].to_pandas()
 - `app.py` - FastAPI application wrapper (can be used as-is).
 - `train.py` - trains and saves a model using the config in the `config/` folder.
 - `config/` - the config file used by `train.py`. 
-- `logs/` - all the logs of `train.py`, the saved model, and predictions on the validation set.
 - `checkpoints/` - the saved model that is used in `predict.py` is here.
 
 - `src/` - Core model & preprocessing logic:

checkpoints/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:39696815c68d20cae615f3e271f5d406d75b866a930851a43f6506f3a593282c
+oid sha256:2e3b2354e9323de072794b7bc4603fee711931b8f9fbbd4e49aab691253bd748
 size 628746

config/config.json

@@ -6,7 +6,7 @@
   "patience": 100,
   "max_epochs": 200,
   "batch_size": 64,
-  "seed": 0,
+  "seed": 42,
   "add_or_mean": "mean",
   "window_size": 15
 }