cyber-security-resources/ai_research/AI Security Best Practices/risk_matrix_example.py

37 lines
1.2 KiB
Python
Raw Normal View History

2023-12-28 19:10:03 -05:00
# A Simple script to illustrate an example of a basic AI Risk Matrix
import matplotlib.pyplot as plt
import numpy as np
# Define the risks and their impact and likelihood
risks = {
"Data Privacy Risk": {"Impact": "Medium", "Likelihood": "Medium"},
"Diagnostic Accuracy Risk": {"Impact": "Very High", "Likelihood": "Low"},
"Bias Risk": {"Impact": "High", "Likelihood": "Medium"}
}
# Mapping of impact and likelihood to numerical values
impact_mapping = {"Low": 1, "Medium": 2, "High": 3, "Very High": 4}
likelihood_mapping = {"Low": 1, "Medium": 2, "High": 3, "Very High": 4}
# Prepare data for plotting
x = [likelihood_mapping[risks[risk]['Likelihood']] for risk in risks]
y = [impact_mapping[risks[risk]['Impact']] for risk in risks]
labels = list(risks.keys())
# Create the plot
plt.figure(figsize=(8, 6))
plt.scatter(x, y, color='blue')
plt.title('AI System Risk Matrix')
plt.xlabel('Likelihood')
plt.ylabel('Impact')
plt.xticks([1, 2, 3, 4], ['Low', 'Medium', 'High', 'Very High'])
plt.yticks([1, 2, 3, 4], ['Low', 'Medium', 'High', 'Very High'])
plt.grid(True)
# Annotate the points
for i, label in enumerate(labels):
plt.annotate(label, (x[i], y[i]))
plt.show()