add experiment

# -*- encoding: utf-8 -*-

'''

@File    :   transis_TDC_to_DTLP.py

@Time    :   2021/09/27 20:55:51

@Author  :   Muyao ZHONG 

@Version :   1.0

@Contact :   zmy125616515@hotmail.com

@License :   (C)Copyright 2019-2020

@Title   :   transis_TDC dataset into DTLP txt data

'''

import pandas as pd 

import numpy as np 

def read_txt(filepath):

    with open(filepath,'r') as f:

        data=pd.read_csv(f,sep='\t')

    return data

def data_pre(data,filename):

    new_data=data[['X1','X2','Y']]

    max_min_scale=lambda x:x/x.max()

    new_data[['Y']]=(new_data[["Y"]]>10000).astype('int')

    # new_data[["Y"]]=new_data[["Y"]]>0.5

    # new_data['Y']=new_data["Y"].astype('int')

    new_data.to_csv(filename,sep=' ',index=None,columns=None)

    return new_data

if __name__ == '__main__':

    data=read_txt("./data/bindingdb_kd.tab")

    data=data_pre(data,"./data/bindingdb_kd.txt")

 No newline at end of file