怎么使用Yellowbrick可视化异常检测模型？-Python教程-PHP中文网

怎么使用Yellowbrick可视化异常检测模型？

看不見的法師

发布： 2025-08-18 19:24:02

原创

494人浏览过

yellowbrick是一个用于异常检测可视化的工具，不是独立算法。1. 选择合适的模型如isolationforest或localoutlierfactor；2. 安装yellowbrick库；3. 准备符合scikit-learn要求的数据集；4. 使用scattervisualizer、rank2d和outlierviz等工具进行可视化；5. 分析图表识别异常模式，结合模型优化参数提升效果。

怎么使用Yellowbrick可视化异常检测模型？

Yellowbrick提供了一套直观的异常检测可视化工具，能帮助你更深入地理解模型表现，并发现数据中的潜在问题。它不是一个独立的异常检测算法，而是各种异常检测算法的可视化辅助工具。

解决方案

选择合适的异常检测模型： Yellowbrick本身不提供异常检测算法，你需要先选择一个合适的模型，例如IsolationForest、LocalOutlierFactor等。 scikit-learn库中包含了这些常用的算法。
安装Yellowbrick： 使用pip安装Yellowbrick库：
```
pip install yellowbrick
```
登录后复制
数据准备： 准备好你的数据集，确保数据格式符合scikit-learn的要求。通常是一个二维数组，每一行代表一个样本，每一列代表一个特征。

使用可视化工具： Yellowbrick提供了多种可视化工具，例如：

散布图矩阵 (Scatter Visualizer)： 可以展示数据集中各个特征之间的关系，帮助你发现异常值可能存在的区域。

from yellowbrick.features import ScatterVisualizer
import pandas as pd
import matplotlib.pyplot as plt

# 假设你有一个名为 'data.csv' 的CSV文件
try:
    data = pd.read_csv('data.csv')
except FileNotFoundError:
    print("错误：找不到名为 'data.csv' 的文件。请确保文件存在并且路径正确。")
    exit()
except pd.errors.EmptyDataError:
    print("错误：'data.csv' 文件为空。请检查文件内容。")
    exit()
except pd.errors.ParserError as e:
    print(f"错误：解析 'data.csv' 文件时发生错误：{e}。请检查文件格式是否正确。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 检查数据是否为空
if data.empty:
    print("错误：加载的数据为空。请检查数据源。")
    exit()

# 移除包含缺失值的行，这是一种常见的处理方式，但请根据你的数据情况选择合适的策略
data = data.dropna()

# 再次检查数据是否为空
if data.empty:
    print("错误：移除缺失值后数据为空。请检查原始数据中缺失值的比例。")
    exit()

# 选择要可视化的特征列。确保这些列存在于你的数据集中
features = data.columns.tolist()

# 检查特征列是否为空
if not features:
    print("错误：没有选择任何特征列。请检查你的数据集中是否有可用的列。")
    exit()

# 创建 ScatterVisualizer 对象
visualizer = ScatterVisualizer(features=features, marker='o', alpha=0.5, size=10)

# 使用数据拟合 visualizer
try:
    visualizer.fit(data.values, data.index)  # 传递数据值和索引
except ValueError as e:
    print(f"ValueError: {e}. 请检查你的数据格式是否正确，以及是否包含非数值数据。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 转换数据
visualizer.transform(data.values)

# 显示可视化结果
visualizer.show()
plt.show()

登录后复制

Rank Features： 用于评估特征的重要性，帮助你选择最相关的特征。

from yellowbrick.features import Rank2D
import pandas as pd
import matplotlib.pyplot as plt

# 假设你有一个名为 'data.csv' 的CSV文件
try:
    data = pd.read_csv('data.csv')
except FileNotFoundError:
    print("错误：找不到名为 'data.csv' 的文件。请确保文件存在并且路径正确。")
    exit()
except pd.errors.EmptyDataError:
    print("错误：'data.csv' 文件为空。请检查文件内容。")
    exit()
except pd.errors.ParserError as e:
    print(f"错误：解析 'data.csv' 文件时发生错误：{e}。请检查文件格式是否正确。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 检查数据是否为空
if data.empty:
    print("错误：加载的数据为空。请检查数据源。")
    exit()

# 移除包含缺失值的行，这是一种常见的处理方式，但请根据你的数据情况选择合适的策略
data = data.dropna()

# 再次检查数据是否为空
if data.empty:
    print("错误：移除缺失值后数据为空。请检查原始数据中缺失值的比例。")
    exit()

# 选择要可视化的特征列。确保这些列存在于你的数据集中
features = data.columns.tolist()

# 检查特征列是否为空
if not features:
    print("错误：没有选择任何特征列。请检查你的数据集中是否有可用的列。")
    exit()

# 创建 Rank2D 对象，method 可以选择 'pearson', 'spearman', 'kendall' 等
visualizer = Rank2D(features=features, algorithm='pearson')

# 使用数据拟合 visualizer
try:
    visualizer.fit(data.values)  # 传递数据值
except ValueError as e:
    print(f"ValueError: {e}. 请检查你的数据格式是否正确，以及是否包含非数值数据。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 转换数据
visualizer.transform(data.values)

# 显示可视化结果
visualizer.show()
plt.show()

登录后复制

OutlierViz： 直接可视化异常检测模型的结果。

from sklearn.ensemble import IsolationForest
from yellowbrick.contrib.classifier import OutlierViz
import pandas as pd
import matplotlib.pyplot as plt

# 假设你有一个名为 'data.csv' 的CSV文件
try:
    data = pd.read_csv('data.csv')
except FileNotFoundError:
    print("错误：找不到名为 'data.csv' 的文件。请确保文件存在并且路径正确。")
    exit()
except pd.errors.EmptyDataError:
    print("错误：'data.csv' 文件为空。请检查文件内容。")
    exit()
except pd.errors.ParserError as e:
    print(f"错误：解析 'data.csv' 文件时发生错误：{e}。请检查文件格式是否正确。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 检查数据是否为空
if data.empty:
    print("错误：加载的数据为空。请检查数据源。")
    exit()

# 移除包含缺失值的行，这是一种常见的处理方式，但请根据你的数据情况选择合适的策略
data = data.dropna()

# 再次检查数据是否为空
if data.empty:
    print("错误：移除缺失值后数据为空。请检查原始数据中缺失值的比例。")
    exit()

# 选择要可视化的特征列。确保这些列存在于你的数据集中
features = data.columns.tolist()

# 检查特征列是否为空
if not features:
    print("错误：没有选择任何特征列。请检查你的数据集中是否有可用的列。")
    exit()

# 初始化异常检测模型
model = IsolationForest()

# 创建 OutlierViz 对象
visualizer = OutlierViz(model, features=features)

# 使用数据拟合 visualizer
try:
    visualizer.fit(data.values)  # 传递数据值
except ValueError as e:
    print(f"ValueError: {e}. 请检查你的数据格式是否正确，以及是否包含非数值数据。")
    exit()
except Exception as e:
    print(f"发生未知错误：{e}。")
    exit()

# 转换数据
visualizer.score(data.values)

# 显示可视化结果
visualizer.show()
plt.show()

登录后复制