import numpy as np import matplotlib import matplotlib.pyplot as plt import p2funcs as p2f vegetables = [ "cucumber", "tomato", "lettuce", "asparagus", "potato", "wheat", "barley" ] farmers = [ "Farmer Joe", "Upland Bros.", "Smith Gardening", "Agrifun", "Organiculture", "BioGoods Ltd.", "Cornylee Corp." ] harvest = np.array([[0.8, 2.4, 2.5, 3.9, 0.0, 4.0, 0.0], [2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0], [1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0], [0.6, 0.0, 0.3, 0.0, 3.1, 0.0, 0.0], [0.7, 1.7, 0.6, 2.6, 2.2, 6.2, 0.0], [1.3, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1], [0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3]]) p2f.plot_mpl_heatmap(harvest, vegetables, farmers, cmap="autumn", cbarlabel="harvest [t/year]", output='show')
opt_t2_gamma = 0 for gamma in t2_gamma_list: amat_dict.update({gamma: []}) t2_auc_mat, t2_kpcas, t2_models = p2f.m_test5_2_rocplot( toy_X_scaled, toy_y, gamma, toy_label, filepath, plotpath, 'tier1') amat_dict[gamma].append(t2_auc_mat) p2f.plot_mpl_heatmap( t2_auc_mat, t2_kpcas, t2_models, cmap="autumn", cbarlabel= "Mean area under ROC curve after 10-fold cross validation", output='save', path='%s%s_tier2_heatmap_gamma_%s.png' % (filepath, nowtime, gamma)) p2f.js_heatmap(t2_auc_mat, t2_kpcas, t2_models, "hmapgamma%s" % (gamma), "%s%sheatmap_gamma%s.js" % (plotpath, nowtime, gamma)) t2_mmas.append(np.mean(t2_auc_mat)) amat_dict_list.append(amat_dict) # Select optimal t1 gamma for i in range(len(t2_mmas)):
print('\n### TIER 1 GRID SEARCH ###') for gamma in t1_gamma_list: amat_dict.update({gamma: []}) t1_auc_mat, t1_kpcas, t1_models = p2f.m_test5_2_rocplot( toy_X_scaled, toy_y, gamma, toy_label, filepath, plotpath, 'tier1') amat_dict[gamma].append(t1_auc_mat) p2f.plot_mpl_heatmap( t1_auc_mat, t1_kpcas, t1_models, cmap="Oranges", cbarlabel= "Mean area under ROC curve after 10-fold cross validation", output='save', path='%s%s_tier1_gs' % (filepath, nowtime)) t1_mmas.append(np.mean(t1_auc_mat)) amat_dict_list.append(amat_dict) # Select optimal t1 gamma for i in range(len(t1_mmas)): if t1_mmas[i] > max_t1_mma: max_t1_mma = t1_mmas[i] opt_t1_gamma = t1_gamma_list[i]