Advertisement
Guest User

KONE

a guest
Jan 18th, 2018
88
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.16 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3. from sklearn.cluster import KMeans
  4. import matplotlib.pyplot as plt
  5. import matplotlib.patches as mpatches
  6. from sklearn.svm import OneClassSVM
  7.  
  8. df = pd.read_csv('Friday.csv') # monday data
  9. df["date"] = np.array(range(len(df["date"])))
  10. df.head()
  11.  
  12. kmeans = KMeans(init='k-means++', n_clusters=3, n_init=7, precompute_distances=True, algorithm='elkan')
  13. preds = kmeans.fit_predict(df)
  14.  
  15. date = df["date"]
  16. vals = df["vals"]
  17. for i in range(3):
  18.     plt.scatter(date[preds == i], vals[preds == i])
  19. plt.show()
  20.  
  21.  
  22. clf = OneClassSVM(nu=0.1, kernel="rbf", gamma=0.001, verbose=True)
  23. for i in range(3):
  24.     date_one = df["date"][preds == i]
  25.     vals_one = df["vals"][preds == i]
  26.     data = np.array([vals_one]).reshape(-1,1)
  27.     clf.fit(data)
  28.     one_preds = clf.predict(data)
  29.     plt.scatter(date_one[one_preds == 1], vals_one[one_preds == 1], c="black")
  30.     plt.scatter(date_one[one_preds == -1], vals_one[one_preds == -1], c="red")
  31. plt.legend(handles=[mpatches.Patch(color='black', label='Normal'), mpatches.Patch(color='red', label='Not Normal')])
  32. plt.xlabel("Dates")
  33. plt.ylabel("Values")
  34. plt.savefig('friday.png')
  35. plt.show()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement