[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-10-30 11:29:10 +00:00
parent fe0cc8dbf8
commit a05f867d5b

View File

@ -11,6 +11,7 @@ from dataclasses import dataclass
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@dataclass @dataclass
class GaussianFuzzySet: class GaussianFuzzySet:
""" """
@ -61,21 +62,30 @@ class GaussianFuzzySet:
>>> GaussianFuzzySet("Medium", 0, 1).complement().membership(0) >>> GaussianFuzzySet("Medium", 0, 1).complement().membership(0)
0.0 0.0
""" """
return GaussianFuzzySet(f"¬{self.name}", self.mean, self.std_dev, is_complement=not self.is_complement) return GaussianFuzzySet(
f"¬{self.name}",
self.mean,
self.std_dev,
is_complement=not self.is_complement,
)
def plot(self): def plot(self):
""" """
Plot the membership function of the Gaussian fuzzy set. Plot the membership function of the Gaussian fuzzy set.
""" """
x = np.linspace(self.mean - 3 * self.std_dev, self.mean + 3 * self.std_dev, 1000) x = np.linspace(
self.mean - 3 * self.std_dev, self.mean + 3 * self.std_dev, 1000
)
y = [self.membership(xi) for xi in x] y = [self.membership(xi) for xi in x]
plt.plot(x, y, label=self.name) plt.plot(x, y, label=self.name)
plt.xlabel("x") plt.xlabel("x")
plt.ylabel("Membership") plt.ylabel("Membership")
plt.legend() plt.legend()
if __name__ == "__main__": if __name__ == "__main__":
from doctest import testmod from doctest import testmod
testmod() testmod()
# Create an instance of GaussianFuzzySet # Create an instance of GaussianFuzzySet
@ -84,7 +94,9 @@ if __name__ == "__main__":
# Display some membership values # Display some membership values
print(f"Membership at mean (25): {fuzzy_set.membership(25)}") print(f"Membership at mean (25): {fuzzy_set.membership(25)}")
print(f"Membership at 30: {fuzzy_set.membership(30)}") print(f"Membership at 30: {fuzzy_set.membership(30)}")
print(f"Complement Membership at mean (25): {fuzzy_set.complement().membership(25)}") print(
f"Complement Membership at mean (25): {fuzzy_set.complement().membership(25)}"
)
# Plot the Gaussian Fuzzy Set and its complement # Plot the Gaussian Fuzzy Set and its complement
fuzzy_set.plot() fuzzy_set.plot()