pyFTS/common/Membership.py

42 lines
1.2 KiB
Python
Raw Normal View History

import numpy as np
import math
from pyFTS import *
def trimf(x, parameters):
xx = round(x, 3)
if (xx < parameters[0]):
return 0
elif (xx >= parameters[0] and xx < parameters[1]):
return (x - parameters[0]) / (parameters[1] - parameters[0])
elif (xx >= parameters[1] and xx <= parameters[2]):
return (parameters[2] - xx) / (parameters[2] - parameters[1])
else:
return 0
def trapmf(x, parameters):
if (x < parameters[0]):
return 0
elif (x >= parameters[0] and x < parameters[1]):
return (x - parameters[0]) / (parameters[1] - parameters[0])
elif (x >= parameters[1] and x <= parameters[2]):
return 1
elif (x >= parameters[2] and x <= parameters[3]):
return (parameters[3] - x) / (parameters[3] - parameters[2])
else:
return 0
def gaussmf(x, parameters):
return math.exp((-(x - parameters[0])**2)/(2 * parameters[1]**2))
#return math.exp(-0.5 * ((x - parameters[0]) / parameters[1]) ** 2)
def bellmf(x, parameters):
return 1 / (1 + abs((x - parameters[2]) / parameters[0]) ** (2 * parameters[1]))
def sigmf(x, parameters):
return 1 / (1 + math.exp(-parameters[0] * (x - parameters[1])))