TwoKey plot v1

examples/visualization_examples/two_key_plot/data_developer_salary_two_key_plot.py

@@ -0,0 +1,39 @@
+from examples.visualization_examples.prepare_datasets import get_data_developer_salary_data
+from niaarm import Dataset, get_rules
+from niaarm.visualize import two_key_plot
+
+# Get prepared data developer salary data
+arm_df = get_data_developer_salary_data()
+
+# Prepare Dataset
+dataset = Dataset(
+    path_or_df=arm_df,
+    delimiter=","
+)
+
+# Get rules
+metrics = ("support", "confidence")
+rules, run_time = get_rules(
+    dataset=dataset,
+    algorithm="DifferentialEvolution",
+    metrics=metrics,
+    max_evals=500
+)
+
+# Sort rules
+rules.sort(by="support")
+# Print rule information
+print("\nRules:")
+print(rules)
+print(f'\nTime to generate rules: {f"{run_time:.3f}"} seconds')
+print("\nRule information: ", rules[3])
+print("Antecedent: ", rules[3].antecedent)
+print("Consequent: ", rules[3].consequent)
+print("Confidence: ", rules[3].confidence)
+print("Support: ", rules[3].support)
+print("Lift: ", rules[3].lift)
+print("\nMetrics:", metrics)
+
+# Visualize scatter plot
+fig = two_key_plot(rules=rules, metrics=metrics, interactive=True)
+fig.show()