main.py 文件源码

def upload_file_kmeans():

    file1 = request.files['trainfile']
    if file1 and allowed_file(file1.filename):
        trainingfilename = secure_filename(file1.filename)
        file1.save(os.path.join(app.config['UPLOAD_FOLDER'], trainingfilename))

    trainingfilename = 'uploads/' + trainingfilename
    my_cluster_df = parse_file(trainingfilename)

    my_cluster_colnames = list(my_cluster_df.columns.values)
    my_cluster_colnames.extend(['cluster'])
    my_cluster_datatypes = []

    list_dataypes = list(my_cluster_df.dtypes)
    for col in list_dataypes:
        if col == 'object':
            my_cluster_datatypes.append('string')
        else:
            my_cluster_datatypes.append('float')
    my_cluster_datatypes.extend(['float'])

    return (json.dumps([my_cluster_colnames, my_cluster_datatypes]))