Advertisement
Guest User

Untitled

a guest
Nov 27th, 2014
141
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.27 KB | None | 0 0
  1. def learn1(data, unique_list):
  2.  
  3. unique_list = []
  4.  
  5. A = data[:, 3]
  6. train_labels = A[:len(A)/2]
  7. test_labels = A[len(A)/2:]
  8. '''
  9. B = unique_list
  10. train_labels = B[:len(B)/2]
  11. test_labels = B[len(B)/2:]
  12. '''
  13. C = data[:, 1]
  14. train_data = C[:len(C)/2]
  15. test_data = C[len(C)/2:]
  16.  
  17. for i in train_data:
  18. if i == 'a':
  19. i = 0
  20. elif i == 'adv':
  21. i = 1
  22. elif i == 'infinitive-marker':
  23. i = 2
  24. elif i == 'det':
  25. i = 3
  26. elif i == 'n':
  27. i = 4
  28. elif i == 'pron':
  29. i = 5
  30. elif i == 'modal':
  31. i = 6
  32. elif i == 'v':
  33. i = 7
  34. elif i == 'conj':
  35. i = 8
  36. elif i == 'prep':
  37. i = 9
  38. elif i == 'interjection':
  39. i = 10
  40. unique_list.append(i)
  41. unique_arr = np.array(unique_list)
  42. print unique_arr
  43. #unique_labels = np.unique(labels)
  44.  
  45. new_arr_list = []
  46. for i in train_data:
  47. temp_list = []
  48. new_arr_list.append(temp_list)
  49. new_arr = np.array(new_arr_list)
  50.  
  51. clf = svm.SVC(kernel = 'linear')
  52. print clf.fit(new_arr, unique_arr)
  53. #return clf.predict(test_data)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement