update plots to reflect epochs used

Makefile

@@ -29,13 +29,13 @@ clean:
 	@rm -rf output/
 	@rm -rf checkpoints/
 
-compress: plots/progress_35845_sm.png plots/progress_680065_sm.png
+compress: plots/progress_35845_sm.png plots/progress_136013_sm.png
 
 plots/progress_35845_sm.png: plots/progress_35845.png
 	@convert -resize 33% plots/progress_35845.png plots/progress_35845_sm.png
 
-plots/progress_680065_sm.png: plots/progress_680065.png
+plots/progress_136013_sm.png: plots/progress_136013.png
-	@convert -resize 33% plots/progress_680065.png plots/progress_680065_sm.png
+	@convert -resize 33% plots/progress_136013.png plots/progress_136013_sm.png
 
 install:
 	pip install -r requirements.txt

README.md

@@ -59,11 +59,11 @@ The approach demonstrated can be extended to other metrics or features beyond ge
 After training, the model should be able to understand the similarity between cities based on their geodesic distances.
 You can inspect the evaluation plots generated by the `eval.py` script to see the improvement in similarity scores before and after training.
 
-After five epochs, the model no longer treats the terms as unrelated:
+Early on in the first epoch, the model no longer treats the terms as totally unrelated:
 ![Evaluation plot](./plots/progress_35845_sm.png)
 
-After ten epochs, we can see the model has learned to correlate our desired quantities:
+After one full epoch, we can see the model has learned to correlate our desired quantities:
-![Evaluation plot](./plots/progress_680065_sm.png)
+![Evaluation plot](./plots/progress_136013_sm.png)
 
 
 *The above plots are examples showing the relationship between geodesic distance and the similarity between the embedded vectors (1 = more similar), for 10,000 randomly selected pairs of US cities (re-sampled for each image).*