У меня проблема с этим сообщением об ошибке. Тип ошибки: неподдерживаемые типы операндов для -: 'str' и 'str' - PullRequest
Новая
       9

У меня проблема с этим сообщением об ошибке. Тип ошибки: неподдерживаемые типы операндов для -: 'str' и 'str'

0 голосов
/ 09 октября 2018

Я пытаюсь реализовать KNN с нуля в python.все функции работали корректно, пока я не дошел до функции main () 

import csv
import random
import math
import operator

with open('iris.csv', 'r') as csvfile:
    lines = csv.reader(csvfile)
    for row in lines:
        print(', '.join(row))

Здесь я создал функцию для чтения файла iris.csv 

def loadDataset(filename, split, trainingSet=[] , testSet=[]):
    with open(filename, 'r') as csvfile:
        lines = csv.reader(csvfile)
        dataset = list(lines)
        for x in range(len(dataset)-1):
            for y in range(4):
                dataset[x][y] = dataset[x][y]
            if random.random() < split:
                trainingSet.append(dataset[x])
            else:
                testSet.append(dataset[x])

Здесь я тестируюфункция 

trainingSet=[]
testSet=[]
loadDataset('iris.csv', 0.66, trainingSet, testSet)
print('Train: ' + repr(len(trainingSet))) 
print('Test: ' + repr(len(testSet)))

Это функция расстояния 

def euclideanDistance(instance1, instance2, length):
    distance = 0
    for x in range(length):
        distance += pow((instance1[x] - instance2[x]), 2)
    return math.sqrt(distance)

Проверка функции расстояния 

data1 = [2, 2, 2, 'a']
data2 = [4, 4, 4, 'b']
distance = euclideanDistance(data1, data2, 3)
print('Distance: ' + repr(distance))

Функция соседей 

def getNeighbors(trainingSet, testInstance, k):
    distances = []
    length = len(testInstance)-1
    for x in range(len(trainingSet)):
        dist = euclideanDistance(testInstance, trainingSet[x], length)
        distances.append((trainingSet[x], dist))
    distances.sort(key=operator.itemgetter(1))
    neighbors = []
    for x in range(k):
    neighbors.append(distances[x][0])
    return neighbors

проверкафункция соседей 

trainSet = [[2, 2, 2, 'a'], [4, 4, 4, 'b']]
testInstance = [5, 5, 5]
k = 3
neighbors = getNeighbors(trainSet, testInstance, 1)
print(neighbors)

функция для получения ответа 

def getResponse(neighbors):
    classVotes = {}
    for x in range(len(neighbors)):
        response = neighbors[x][-1]
        if response in classVotes:
            classVotes[response] += 1
        else:
            classVotes[response] = 1
    sortedVotes = sorted(classVotes.items(), key=operator.itemgetter(1), reverse=True)
    return sortedVotes[0][0]

проверка функции 

neighbors = [[1,1,1,'a'], [2,2,2,'a'], [3,3,3,'b']]
response = getResponse(neighbors)
print(response)

функция точности 

def getAccuracy(testSet, predictions):
    correct = 0
    for x in range(len(testSet)):
        if testSet[x][-1] is predictions[x]:
            correct += 1
    return (correct/float(len(testSet))) * 100.0

проверкафункция точности 

testSet = [[1,1,1,'a'], [2,2,2,'a'], [3,3,3,'b']]
predictions = ['a', 'a', 'a']
accuracy = getAccuracy(testSet, predictions)
print(accuracy)

Здесь идет main (), где лежит ошибка 

#main
def main():
    # prepare data
    trainingSet=[]
    testSet=[]
    split = 0.67
    loadDataset('iris.csv', split, trainingSet, testSet)
    print ('Train set: ' + repr(len(trainingSet)))
    print ('Test set: ' + repr(len(testSet)))
    # generate predictions
    predictions=[]
    k = 3
    for x in range(len(testSet)):
        neighbors = getNeighbors(trainingSet, testSet[x], k)
        result = getResponse(neighbors)
        predictions.append(result)
        print('> predicted=' + repr(result) + ', actual=' + repr(testSet[x][-1]))
    accuracy = getAccuracy(testSet, predictions)
    print('Accuracy: ' + repr(accuracy) + '%')

main()

, но когда я добираюсь до функции main (), я получаю это сообщение об ошибке 

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-68-ba4ef0fcfe9f> in <module>()
     18     print('Accuracy: ' + repr(accuracy) + '%')
     19 
---> 20 main()

<ipython-input-68-ba4ef0fcfe9f> in main()
     11     k = 3
     12     for x in range(len(testSet)):
---> 13         neighbors = getNeighbors(trainingSet, testSet[x], k)
     14         result = getResponse(neighbors)
     15         predictions.append(result)

<ipython-input-62-d13cbe2070b3> in getNeighbors(trainingSet, testInstance, k)
      3     length = len(testInstance)-1
      4     for x in range(len(trainingSet)):
----> 5         dist = euclideanDistance(testInstance, trainingSet[x], length)
      6         distances.append((trainingSet[x], dist))
      7     distances.sort(key=operator.itemgetter(1))

<ipython-input-60-93ee3f7cf267> in euclideanDistance(instance1, instance2, 
length)
      2     distance = 0
      3     for x in range(length):
----> 4         distance += pow(float(instance1[x] - instance2[x]), 2)
      5     return math.sqrt(distance)

TypeError: unsupported operand type(s) for -: 'str' and 'str'

1 Ответ

0 голосов
/ 09 октября 2018

Вы опубликовали слишком много кода, но, к счастью, сообщение об ошибке и трассировка стека понятны: вы пытаетесь «вычесть» две вещи, которые являются строками, а не числами.Пакет csv вернет все как строку, поэтому вам нужно явно преобразовать числовые столбцы в int (или float) в результате, прежде чем вы сможете рассматривать их как числа.

...