logistic_regression.py

#import libraries needed
import numpy as np
from sklearn import linear_model
if __name__=="__main__":
	#load Dataset
	#here i am creating one for Single variable Linear regression
	x = np.array([[1, 2], [1, 4], [2, 3], [2, 2], [4.5, 7]])
	y = np.array([0, 0, 1, 0, 1])

	#Create an object for logistic regression
	logistic = linear_model.LogisticRegression()
	logistic.fit(x, y)

	#predict for input
	print(logistic.predict([[2, 3]]))