

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier

# Load iris file
_a:      Tuple								=     load_iris()
iris.keys()


print(F'Target names: \n {iris.target_names} ')
print(F'\n Features: \n {iris.feature_names}')

# Train set e Test set
_a							,     _a							,     _a							,     _a:      str								=     train_test_split(
    iris["""data"""], iris["""target"""], random_state=4
)

# KNN

_a:      str								=     KNeighborsClassifier(n_neighbors=1)
knn.fit(X_train, y_train)

# new array to test
_a:      List[str]								=     [[1, 2, 1, 4], [2, 3, 4, 5]]

_a:      str								=     knn.predict(X_new)

print(
    F'\nNew array: \n {X_new}\n\nTarget Names Prediction: \n'
    F' {iris["target_names"][prediction]}'
)


from cva import COLOR_BGR2GRAY, cvtColor, imread, imshow, waitKey
from numpy import divide, inta, multiply, ravel, sort, zeros_like



def  __lowerCAmelCase (  A    ,       A=3      ):
	UpperCAmelCase_    			=     int(mask / 2      )
	# copy image size
	UpperCAmelCase_    			=     zeros_like(A      )
	for i in range(A    ,       gray_img.shape[0] - bd      ):
		for j in range(A    ,       gray_img.shape[1] - bd      ):
			# get mask according with mask
			UpperCAmelCase_    			=     ravel(gray_img[i - bd : i + bd + 1, j - bd : j + bd + 1]      )
			# calculate mask median
			UpperCAmelCase_    			=     sort(A      )[inta(divide((multiply(A    ,       A      ))    ,       2      ) + 1      )]
			UpperCAmelCase_    			=     median
	return median_img


if __name__ == "__main__":
				# read original image
				_a:      Optional[int]								=     imread("""../image_data/lena.jpg""")
				# turn image in gray scale value
				_a:      Dict								=     cvtColor(img, COLOR_BGR2GRAY)

				# get values with two different mask size
				_a:      Tuple								=     median_filter(gray, 3)
				_a:      Dict								=     median_filter(gray, 5)

				# show result images
				imshow("""median filter with 3x3 mask""", medianaxa)
				imshow("""median filter with 5x5 mask""", medianaxa)
				waitKey(0)