ValueError: could not convert string to float

weixin_40313521

于 2019-03-08 13:53:39 发布

阅读量4.1k

点赞数 1

文章标签：数据分析 python 机器学习 could not convert string to float

本文链接：https://blog.csdn.net/weixin_40313521/article/details/88350224

版权

ValueError: could not convert string to float

from sklearn import svm
from sklearn.model_selection import train_test_split
from sklearn import metrics
from sklearn.preprocessing import StandardScaler

def read_data(data_path):
    with open(data_path, 'r') as f:
        data = []
        for line in f.readlines():
            data.append(line.strip())
    return data


features = read_data(r"features.dat")
labels = read_data(r"labels.dat")

train_features, test_features, train_labels, test_labels = train_test_split(features, labels, test_size=0.33,
                                                                            random_state=33)

scaler = StandardScaler()
train_features = scaler.fit_transform(train_features)
test_features = scaler.fit_transform(test_features)

model = svm.SVC()
model.fit(train_features, train_labels)

prediction =