diff --git a/README.md b/README.md
index 273fc79..8239898 100644
--- a/README.md
+++ b/README.md
@@ -2,12 +2,12 @@
[](LICENSE)
[](pyproject.toml)
+[](https://pypi.org/project/dpg/)
[](https://github.com/Meta-Group/DPG/actions/workflows/ci.yml)
[](https://dpg.readthedocs.io/en/latest/)
-
-
-
+|  |
+|:--:|
DPG is a model-agnostic framework for global and local explanation of tree-based ensemble models.
@@ -20,9 +20,7 @@ insightful points. DPG enables graph-based evaluations and the identification of
towards facilitating comparisons between features and their associated values while offering insights
into the entire model. DPG provides descriptive metrics that enhance the understanding of the
decisions inherent in the model, offering valuable insights.
-
-
-
+
---
@@ -48,9 +46,7 @@ The concept behind DPG is to convert a generic tree-based ensemble model for cla
- Nodes represent predicates, i.e., the feature-value associations present in each node of every tree;
- Edges denote the frequency with which these predicates are satisfied during the model training phase by the samples of the dataset.
-
-
-
+
## Metrics
The graph-based nature of DPG provides significant enhancements in the direction of a complete mapping of the ensemble structure.
@@ -318,20 +314,14 @@ Some examples can be appreciated in the `examples` folder: https://github.com/Me
In particular, the following DPG is obtained by transforming a Random Forest with 5 base learners, trained on Iris dataset.
The used command is `python run_dpg_standard.py --dataset iris --n_learners 5 --pv 0.001 --t 2 --dir examples --plot --save_plot_dir examples`.
-
-
-
+
The following visualizations are obtained using the same parameters as the previous example, but they show two different metrics: _Community_ and _Betweenness centrality_.
The used command for showing communities is `python run_dpg_standard.py --dataset iris --n_learners 5 --pv 0.001 --t 2 --dir examples --plot --save_plot_dir examples --communities`.
-
-
-
+
The used command for showing a specific property is `python run_dpg_standard.py --dataset iris --n_learners 5 --pv 0.001 --t 2 --dir examples --plot --save_plot_dir examples --attribute "Betweenness centrality" --class_flag`.
-
-
-
+
***
## Citation