You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
908 B
29 lines
908 B
from sklearn.neighbors import KNeighborsClassifier
|
|
from sklearn.linear_model import LogisticRegression
|
|
from sklearn.tree import DecisionTreeClassifier
|
|
from sklearn.linear_model import SGDClassifier
|
|
|
|
def logistic_regression(X_train, y_train, X_test):
|
|
logistic = LogisticRegression(max_iter = 100000)
|
|
logistic.fit(X_train,y_train)
|
|
|
|
return logistic.predict(X_test), logistic
|
|
|
|
def decision_tree(X_train, y_train, X_test):
|
|
decisionTree = DecisionTreeClassifier()
|
|
decisionTree = decisionTree.fit(X_train,y_train)
|
|
|
|
return decisionTree.predict(X_test), decisionTree
|
|
|
|
def knn_classifier(X_train, y_train, X_test):
|
|
knn = KNeighborsClassifier(n_neighbors=5)
|
|
knn.fit(X_train, y_train)
|
|
|
|
return knn.predict(X_test), knn
|
|
|
|
def sgd_classifier(X_train, y_train, X_test):
|
|
sgd = SGDClassifier(loss="hinge", penalty="l2")
|
|
sgd.fit(X_train, y_train)
|
|
|
|
return sgd.predict(X_test), sgd
|