Add files via upload

Author: geekquad

K means.py

@@ -0,0 +1,58 @@
+
+# coding: utf-8
+
+# In[1]:
+
+
+import pickle
+import numpy
+import sys
+import matplotlib.pyplot as plt
+sys.path.append(".../tools/")
+from feature_format import featureFormat, targetFeatureSplit
+
+
+# In[5]:
+
+
+def Draw(pred, features, poi, mark_poi=False, name='image.png', f1_name='feature 1', f2='feature 2'):
+ colors = ["b", "c", "k", "m", "g"]
+ for ii, pp in enumerate(pred):
+ plt.scatter(features[ii][0], features[ii][1], color= colors[pred[ii]])
+ 
+ if mark_poi:
+ for ii, pp in enumerate(pred):
+ if poi[ii]:
+ plt.scatter(features[ii][0], features[ii][1], color="r", marker="*")
+ 
+ plt.xlabel(f1_name)
+ plt.ylabel(f2_name)
+ plt.savefig(name)
+ plt.show()
+
+
+# In[9]:
+
+
+data_dict = pickle.load(open("C:/Users/Geekquad/final_project_dataset_modified_unix.pkl", "rb"))
+data_dict.pop("TOTAL", 0)
+
+
+# In[10]:
+
+
+feature_1 = 'salary'
+feature_2 = "exercised_stock_options"
+poi = "poi"
+features_list = [poi, feature_1, feature_2]
+data = featureFormat(data_dict, features_list)
+poi, finance_features = targetFeatureSplit(data)
+
+
+# In[11]:
+
+
+for f1, f2 in finance_features:
+ plt.scatter(f1, f2)
+plt.show()
+