mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
"""Unit tests for Anomalib's Gaussian Mixture Model."""
|
|
|
|
# Copyright (C) 2022-2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import logging
|
|
|
|
import torch
|
|
from _pytest.logging import LogCaptureFixture
|
|
|
|
from anomalib.models.components.cluster.gmm import GaussianMixture
|
|
|
|
|
|
def test_fit_and_predict() -> None:
|
|
"""Test that the model can fit to some synthetic data and correctly predict the labels."""
|
|
# Create a GaussianMixture model
|
|
model = GaussianMixture(n_components=2)
|
|
|
|
# Create some synthetic data
|
|
data = torch.cat(
|
|
[
|
|
torch.randn(100, 2) + torch.tensor([10.0, 10.0]),
|
|
torch.randn(100, 2) + torch.tensor([-10.0, -10.0]),
|
|
],
|
|
)
|
|
|
|
# Fit the model to the data
|
|
model.fit(data)
|
|
|
|
# Predict the labels of the data
|
|
labels = model.predict(data)
|
|
|
|
# Check that the labels are as expected
|
|
assert (labels[:100] == labels[0]).all()
|
|
assert (labels[100:] == labels[100]).all()
|
|
assert (labels[:100] != labels[100:]).all()
|
|
|
|
|
|
def test_warns_on_non_convergence(caplog: LogCaptureFixture) -> None:
|
|
"""Test that the model warns when it does not converge."""
|
|
# Create a GaussianMixture model
|
|
model = GaussianMixture(n_components=2, n_iter=1)
|
|
|
|
# Create some synthetic data
|
|
data = torch.cat(
|
|
[
|
|
torch.randn(100, 2) + torch.tensor([2.0, 2.0]),
|
|
torch.randn(100, 2) + torch.tensor([-2.0, -2.0]),
|
|
],
|
|
)
|
|
|
|
# Fit the model to the data
|
|
with caplog.at_level(logging.WARNING):
|
|
model.fit(data)
|
|
|
|
assert "GMM did not converge" in caplog.text
|