Advertisement
Guest User

Untitled

a guest
Dec 6th, 2019
118
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.91 KB | None | 0 0
  1. def scale_data(data):
  2.     return MinMaxScaler().fit_transform(data[0]), data[1]
  3.  
  4. def prepare_cancer_data(data):
  5.     points, labels = data
  6.     return points, np.array([1 if label == 'M' else 0 for label in labels])
  7.  
  8. def prepare_sms_data(data):
  9.     points, labels = data
  10.     return points, np.array([1 if label == 'ham' else 0 for label in labels])
  11.  
  12. def data_split(data, test_size=0.2):
  13.     points, labels = data
  14.     train_points, test_points, train_labels, test_labels = train_test_split(points, labels, test_size=0.2)
  15.     return (train_points, train_labels), (test_points, test_labels)
  16.  
  17.  
  18. spam_data = scale_data(spam_data)
  19. sms_data = prepare_sms_data(sms_data)
  20. cancer_data = prepare_cancer_data(scale_data(cancer_data))
  21.  
  22. train_blob, test_blob = data_split(blob_data)
  23. train_cancer, test_cancer = data_split(cancer_data)
  24. train_spam, test_spam = data_split(spam_data)
  25. train_sms, test_sms = data_split(sms_data)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement