diff --git a/README.md b/README.md index 854f165d..6814bc16 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,5 @@ -One two three -./`download-dataset` +First download existing dataset with `download-dataset`. +Run `train-embedding.py` to train embedding model with `embedding_config.json`. If `visualize` is set to `true` will create loss graph and PCA+TSNE visualization. +Run `train-classifier.py` to train classification (kNN) model with `classifier_config.json` + +