Matan Kriel commited on
Commit
25168f2
·
1 Parent(s): 879ede5

added plot to readme

Browse files
Files changed (1) hide show
  1. README.md +12 -3
README.md CHANGED
@@ -40,9 +40,18 @@ We compared three algorithms to find the best predictor:
40
  * **MAPE**: Average percentage error.
41
 
42
  ### 3. Advanced Analysis (Plots)
43
- * **`diversity_plot.png`**: A PCA visualization showing the semantic spread of the 10,000 generated descriptions.
44
- * **`model_comparison.png`**: Bar chart comparing RMSE across models and ROC curves for viral classification validity.
45
- * **`feature_importance.png`**: The top 20 words and metadata features that drive virality in our simulated world.
 
 
 
 
 
 
 
 
 
46
 
47
  ## 🛠️ Tech Stack
48
  * **Core**: Python, Pandas, Numpy, Scikit-Learn
 
40
  * **MAPE**: Average percentage error.
41
 
42
  ### 3. Advanced Analysis (Plots)
43
+
44
+ #### Semantic Diversity (PCA)
45
+ ![Diversity Plot](project_plots/diversity_plot.png)
46
+ *A PCA visualization showing the semantic spread of the 10,000 generated descriptions.*
47
+
48
+ #### Model Performance
49
+ ![Model Comparison](project_plots/model_comparison.png)
50
+ *Bar chart comparing RMSE across models and ROC curves for viral classification validity.*
51
+
52
+ #### Feature Importance
53
+ ![Feature Importance](project_plots/feature_importance.png)
54
+ *The top 20 words and metadata features that drive virality in our simulated world.*
55
 
56
  ## 🛠️ Tech Stack
57
  * **Core**: Python, Pandas, Numpy, Scikit-Learn