search

UMass-Rescue / CombinedTechStack

Handle and process large amounts of media data with plug-and-play machine learning models
https://umass-rescue.github.io/CombinedTechStack/
MIT License
1 stars 1 forks source link

Handle Automatic Train/Test Split #14

Open CodyRichter opened 3 years ago

CodyRichter commented 3 years ago

Right now the server only supports a train/validation split with the entire dataset, and no test set is generated when the dataset is loaded.

Create a new environment variable in docker-compose.yml for train/test split, and then use this new test set to provide additional statistics on model training.