import numpy as np
from sklearn.cluster import KMeans
import numpy as np
from sklearn.metrics import silhouette_score, davies_bouldin_score


datatypes=["equal","heterogeneous"]
typenums=[5,10,15]
data_names = ['VAE_FCTAE_EM','AE_FAETC_EM', 'AE_FCTAE_EM', 'DAE_FAETC_EM', 'DAE_FCTAE_EM','SVAE_FCTAE_EM','MMDVAE_EM']
for datatype in datatypes:
    for typenum in typenums:
        for data_name in data_names:
            encoded_factors=np.loadtxt('./result/simulations/{datatype}/{typenum}/{d}_{typenum}.txt'.format(datatype=datatype,typenum=typenum,d=data_name))
            savepath='./result/simulations/{datatype}/{typenum}/{d}_cluster_result.txt'.format(datatype=datatype,typenum=typenum,d=data_name)
            with open(savepath, 'w') as f2:
                print('method:{d}\n'.format(d=data_name))
                f2.write('method:{d}\n'.format(d=data_name))
                for cluster_num in range(2,16,1):
                    all_silhouette=[]
                    all_DBI=[]
                    for i in range(100):
                        clf = KMeans(n_clusters=cluster_num)
                        clf.fit(encoded_factors)  # 模型训练
                        labels = clf.labels_
                        silhouetteScore = silhouette_score(encoded_factors, labels, metric='euclidean')
                        all_silhouette.append(silhouetteScore)
                        davies_bouldinScore = davies_bouldin_score(encoded_factors, labels)
                        all_DBI.append(davies_bouldinScore)
                    avg_silhouette=np.mean(all_silhouette)
                    avg_DBI=np.mean(all_DBI)

                    # print("silhouetteScore:", avg_silhouette)
                    # print("davies_bouldinScore:", avg_DBI)
                    print('k:{k}\nsilhouetteScore:{s}\ndavies_bouldinScore:{d}\n'.format(k=cluster_num, s=avg_silhouette,d=avg_DBI))
                    f2.write('*'*20+'\n')
                    f2.write('k:{k}\nsilhouetteScore:{s}\ndavies_bouldinScore:{d}\n'.format(k=cluster_num, s=avg_silhouette,d=avg_DBI))


#直接拼接
# files = ['aml', 'breast', 'colon', 'kidney', 'liver', 'lung', 'melanoma', 'ovarian', 'sarcoma','gbm']
# for f in files:
#     datapath='./data/cancer_do_cluster/{f}'.format(f=f)
#     omics1 = np.loadtxt('{}/log_exp_omics.txt'.format(datapath))
#     omics1 = np.transpose(omics1)
#     omics1 = normalize(omics1, axis=0, norm='max')
#     print(omics1.shape)
#     omics2 = np.loadtxt('{}/log_mirna_omics.txt'.format(datapath))
#     omics2 = np.transpose(omics2)
#     omics2 = normalize(omics2, axis=0, norm='max')
#     print(omics2.shape)
#     omics3 = np.loadtxt('{}/methy_omics.txt'.format(datapath))
#     omics3 = np.transpose(omics3)
#     omics3 = normalize(omics3, axis=0, norm='max')
#     print(omics3.shape)
#     omics = np.concatenate((omics1, omics2, omics3), axis=1)
#     encoded_factors=omics
#     savepath='./result/cancer_do_cluster/{f}/Contact_cluster_result.txt'.format(f=f)
#     with open(savepath, 'w') as f2:
#         print('cancer:{f}\nmethod:直接拼接'.format(f=f))
#         f2.write('cancer:{f}\nmethod:直接拼接\n'.format(f=f))
#         for typenum in range(2,7,1):
#             all_silhouette=[]
#             all_DBI=[]
#             for i in range(100):
#                 clf = KMeans(n_clusters=typenum)
#                 clf.fit(encoded_factors)  # 模型训练
#                 labels = clf.labels_
#                 silhouetteScore = silhouette_score(encoded_factors, labels, metric='euclidean')
#                 all_silhouette.append(silhouetteScore)
#                 davies_bouldinScore = davies_bouldin_score(encoded_factors, labels)
#                 all_DBI.append(davies_bouldinScore)
#             avg_silhouette=np.mean(all_silhouette)
#             avg_DBI=np.mean(all_DBI)

#             # print("silhouetteScore:", avg_silhouette)
#             # print("davies_bouldinScore:", avg_DBI)
#             print('k:{k}\nsilhouetteScore:{s}\ndavies_bouldinScore:{d}\n'.format(k=typenum, s=avg_silhouette,d=avg_DBI))
#             f2.write('zly'*20+'\n')
#             f2.write('k:{k}\nsilhouetteScore:{s}\ndavies_bouldinScore:{d}\n'.format(k=typenum, s=avg_silhouette,d=avg_DBI))