pyFTS/partitioners/Util.py
Petrônio Cândido de Lima e Silva 17e13fae0b Particionamento com a Gaussmf
2016-12-26 12:46:29 -02:00

28 lines
967 B
Python

import numpy as np
import pandas as pd
import matplotlib as plt
import matplotlib.colors as pltcolors
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from pyFTS.common import Membership
def plotSets(data, sets,num):
fig = plt.figure(figsize=[20, 10])
maxx = max(data)
minx = min(data)
h = 1/num
for k in range(num):
ax0 = fig.add_axes([0, (k+1)*h, 0.65, h]) # left, bottom, width, height
ax0.set_ylim([0, 1])
ax0.set_xlim([minx, maxx])
for s in sets[k]:
if s.mf == Membership.trimf:
ax0.plot([s.parameters[0],s.parameters[1],s.parameters[2]],[0,1,0])
elif s.mf == Membership.gaussmf:
print( s.parameters )
print([s.lower, s.upper])
tmpx = [ kk for kk in np.arange(s.lower, s.upper)]
tmpy = [s.membership(kk) for kk in np.arange(s.lower, s.upper)]
ax0.plot(tmpx, tmpy)