def main(trainset, window, nfeatures, image, savetodir):
output = savetodir+"temp.xpl"
#building the xpl file
ensemble.build_xpl(trainset,window,output)
#reading the "just" created xpl file
result = xplutil.read_xpl(output)
XPL_data = result.data
w0 = result.freq0.copy()
w1 = result.freq1.copy()
#normalizing the table frequency values
w0,w1 = cl.normalize_table(w0, w1)
#calculating features and indexes
indices, feature_list, _ = ft.cmim(XPL_data, w0, w1, nfeatures)
#saving window file
tw.to_window_file(indices, result.winshape, savetodir+"window.win")
#applying the feature selection algorithm
w0, w1, updated_decision, cls_error = cl.apply_feature_selection(XPL_data, indices, w0, w1)
#calculating the unique array and their indexes
indices = np.sort(indices)
unique_array, unique_index = cl._apply_projection(XPL_data, indices)
#writing a mimterm file to disk
xplutil.write_minterm_file(savetodir+"mintermfile.mtm",indices, result.winshape, unique_array,updated_decision[unique_index])
#building the operator based on the mintermfile
ensemble.build_operator(savetodir+"window.win", savetodir+"mintermfile.mtm", savetodir+"operator")
#building a new XPL according to the learned window
output = savetodir+"Learned.xpl"
ensemble.build_xpl(trainset,savetodir+"window.win",output)
result_img = savetodir+"Image_applied"
#applying the operator on the image
ensemble.apply_operator(savetodir+"operator", image, result_img)
def main(xpl_data, num_features=None, plotwindow=False, winfile=None):
height, width = xpl_data.winshape
indices, feature_list, _ = feature.cmim(xpl_data.data, xpl_data.freq0,
xpl_data.freq1, num_features)
print "Selected features: ", indices
if plotwindow:
trioswindow.plot_pixels(indices, xpl_data.winshape)
if winfile:
trioswindow.to_window_file(indices, xpl_data.winshape, winfile)
def make_windows(savetodir, winshape, m, n_windows):
"""
a function to make the windows randomly and save to dir
savetodir is the directory to save the windows
winshape is the window shape
m is the number of pixels to be selected
n_windows is the number of windows to save
"""
for i in range(int(n_windows)):
pixels = _get_indexes(m, winshape)
tw.to_window_file(pixels, winshape, savetodir + "window_" + str(i) + ".win")
tw.to_image_file(pixels, winshape, savetodir + "window_" + str(i) + ".png", scale=8)
return 0
def main(xpl_data, num_features=None, num_iterations=None, save_todir=None):
data = xpl_data.data
# copies frequency data as original frequencies are used towards the end to estimate training error
w0 = xpl_data.freq0.copy()
w1 = xpl_data.freq1.copy()
error_list = []
mae_list = []
DEC = np.zeros(w0.shape)
GVector = []
i=0
winfile = "window_"
win = ".win"
png = ".png"
w0_train, w1_train = cl.normalize_table(xpl_data.freq0, xpl_data.freq1)
file = open(save_todir+"Error.txt", "w")
for i in range(num_iterations):
indices, feature_list, _ = ft.cmim(data, w0, w1, num_features)
tw.to_window_file(indices, xpl_data.winshape, save_todir+winfile+str(i)+win)
tw.to_image_file(indices,xpl_data.winshape, save_todir+winfile+str(i)+png, scale=8)
w0, w1 = cl.normalize_table(w0, w1)
w0, w1, updated_decision, cls_error = cl.apply_feature_selection(data, indices, w0, w1)
unique_array, unique_index = cl._apply_projection(data, indices)
xplutil.write_minterm_file(save_todir+"mtm_"+str(i),indices, xpl_data.winshape,unique_array,updated_decision[unique_index])
str_to_file = "Classification error for iteration " + str(i) +" = "+ str(cls_error) +".\n"
file.write(str_to_file)
error_list.append(cls_error)
bt = cl.beta_factor(cls_error)
gam = np.log(1/bt)
GVector = np.append(GVector,gam)
#DEC represents the Decision Table. Each column represents the decision
#for an iteration
DEC = np.column_stack((DEC,updated_decision))
aux_dec = DEC
aux_dec = np.delete(aux_dec,0, axis=1)
hypothesis = cl.adaboost_decision(aux_dec, GVector)
MAE_t = cl.mae_from_distribution(hypothesis,w0_train, w1_train)
mae_list = np.append(mae_list,MAE_t)
str_to_file = "MAE for iteration " + str(i) +" = "+ str(MAE_t) +".\n\n"
file.write(str_to_file)
#Must delete the first column because it contains only Zeros as it was initialized with np.zeros()
DEC = np.delete(DEC,0, axis=1)
hypothesis = cl.adaboost_decision(DEC, GVector)
MAE = cl.mae_from_distribution(hypothesis, w0_train, w1_train)
str_to_file = "Final MAE = "+ str(MAE)
file.write(str_to_file)
#print MAE
file.close()
gra.plot_MAE_iter(np.array(range(num_iterations)), np.array(mae_list))
def train(xpl_data, n_features, n_iterations, dirpath):
Xdata = xpl_data.data
win = xpl_data.windata
# copies frequency data as original frequencies are used towards the end to estimate training error
w0 = xpl_data.freq0.copy()
w1 = xpl_data.freq1.copy()
error_list = []
mae_list = []
GVector = []
DEC = np.zeros(w0.shape)
total = float(np.sum([w0, w1]))
w0_train = w0/total
w1_train = w1/total
file = open(dirpath+"MAE_training.txt", "w")
for i in range(n_iterations):
indices, feature_list, _ = ft.cmim(Xdata, w0, w1, n_features)
indices = np.sort(indices)
triosw.to_window_file(indices, xpl_data.winshape, dirpath+"window_"+str(i)+".win")
triosw.to_image_file(indices,xpl_data.winshape, dirpath+"window_"+str(i)+".png", scale=8)
total = float(np.sum([w0, w1]))
w0 = w0/total
w1 = w1/total
w0, w1, updated_decision, cls_error = clf.apply_feature_selection(Xdata, indices, w0, w1)
unique_array, unique_index = clf._apply_projection(Xdata, indices)
xplutil.write_minterm_file(dirpath+"mtm"+str(i),indices, xpl_data.winshape, unique_array,updated_decision[unique_index])
#str_to_file = "Classification error for iteration " + str(i) +" = "+ str(cls_error) +".\n"
#file.write(str_to_file)
error_list.append(cls_error)
bt = clf.beta_factor(cls_error)
gam = np.log(1/bt)
GVector = np.append(GVector,gam)
#DEC represents the Decision Table. Each column represents the decision for an iteration
DEC = np.column_stack((DEC,updated_decision))
aux_dec = DEC
aux_dec = np.delete(aux_dec,0, axis=1)
hypothesis = clf.adaboost_decision(aux_dec, GVector)
MAE_t = clf.mae_from_distribution(hypothesis,w0_train, w1_train)
mae_list = np.append(mae_list,MAE_t)
str_to_file = str(i) +", "+ str(MAE_t) +"\n"
file.write(str_to_file)
#Must delete the first column because it contains only Zeros as it was initialized with np.zeros()
DEC = np.delete(DEC,0, axis=1)
hypothesis = clf.adaboost_decision(DEC, GVector)
#MAE = clf.mae_from_distribution(hypothesis,w0_train, w1_train)
#str_to_file = "Final MAE = "+str(MAE)
#file.write(str(MAE))
file.close()
plot_MAE(np.array(range(n_iterations)), np.array(mae_list), dirpath)
return Ensemble(xpl_data, win, n_features, n_iterations, error_list, mae_list,dirpath)
