helperFuncs.py 文件源码

python
阅读 27 收藏 0 点赞 0 评论 0

项目:PersonalizedMultitaskLearning 作者: mitmedialab 项目源码 文件源码
def generateWekaFile(X,Y,features,path,name):
    f = open(path + name + '.arff', 'w')
    f.write("@relation '" + name + "'\n\n")

    for feat in features:
        f.write("@attribute " + feat + " numeric\n")
    f.write("@attribute cluster {True,False}\n\n")

    f.write("@data\n\n")
    for i in range(X.shape[0]):
        for j in range(X.shape[1]):
            if np.isnan(X[i,j]):
                f.write("?,")
            else:
                f.write(str(X[i,j]) + ",")
        if Y[i] == 1.0 or Y[i] == True:
            f.write("True\n")
        else:
            f.write("False\n")

    f.close()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号