使用map作为随机森林分类器的输入-Java 学习之路

我在python中使用sklearn模块时遇到问题 . 我正在尝试使用随机森林分类器使用以下代码对16个不同类进行分类：

count = 0
for line in lines:
    tokens = line.strip().split(",")
    if count < TRAINING_SIZE:
        X.append(map(float, tokens[0:NUM_PCA]))
        if CLASSIFICATION_TYPE == 2:
            y.append(int(tokens[len(tokens)-1]))
        elif int(tokens[len(tokens)-1]) == 1:
            y.append(0)
        else:
            y.append(1)
        count += 1
    else:
        test_X.append(map(float, tokens[0:NUM_PCA]))
        if CLASSIFICATION_TYPE == 2:
            test_y.append(int(tokens[len(tokens)-1]))
        elif int(tokens[len(tokens)-1]) == 1:
            test_y.append(0)
        else:
            test_y.append(1)

clf = RandomForestClassifier(n_estimators=10, max_depth=None, min_samples_split=1, random_state=0)
clf.fit(X,y)

train_predictions = clf.predict(X)
test_predictions = clf.predict(test_X)

但它返回一个错误，表明：

TypeError：float（）参数必须是字符串或数字，而不是'map'

如何在此功能中使用 Map ？

1 回答

0
如果你改变：
```
X.append(map(float, tokens[0:NUM_PCA]))
```
至：
```
X.append([float(tok) for tok in tokens[0:NUM_PCA]])
```
这个问题应该修复 .

请注意，您必须对 test_X.append(...) 执行相同的操作
回复于 2024-05-05T17:44:20+08:00

使用map作为随机森林分类器的输入

1 回答

相关问题