diff options
Diffstat (limited to 'data/svm')
| -rwxr-xr-x | data/svm/classification.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/data/svm/classification.py b/data/svm/classification.py index 26cb283..08e5363 100755 --- a/data/svm/classification.py +++ b/data/svm/classification.py @@ -43,10 +43,10 @@ def normalize_filter(a,weights=None,nameset=None): a,weights = normalize(a,weights) if nameset != None: - indexes = [i for i in range(a.shape[0]) if a[i][0] in nameset] + indexes = [i for i in range(a.shape[0]) if a[i,0] in nameset] a = a[indexes] - return list(a[:,0]),[{i:v for i,v in enumerate(row[4:])} for row in a],weights + return list(a[:,0]),[{i+1:v for i,v in enumerate(row[4:])} for row in a],weights def perform_svm(a,b,nameset=None): y1,x1,weights = normalize_filter(a,nameset=nameset) |