search

nosqlgeek / ai-meets-museum

The idea of this project is to develop an AI-based tool that allows easier information gathering around artifacts in a museum.
BSD 3-Clause "New" or "Revised" License
2 stars 0 forks source link

python code to split Dataset for training model #21

Open stephanschorer opened 1 year ago

stephanschorer commented 1 year ago

Split Dataset into testing and training data

Paused until Model needs to be retrained

ToDos

stephanschorer commented 1 year ago

we did not retrain our model, but that could be a good starting point in the future

there is already some code - look at https://github.com/nosqlgeek/ai-meets-museum/blob/imageClassification/src/tools/imageClassification/splitDataset.py https://github.com/nosqlgeek/ai-meets-museum/blob/imageClassification/src/tools/imageClassification/trainModel.py