sklearn

Adding sklearn on Function Details Page

Figure 1 Adding sklearn

Importing sklearn to Code

# Import sklearn.
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
def handler (event, context):
       iris=datasets.load_iris()
       iris_X=iris.data
       iris_y=iris.target
    
       X_train,X_test,y_train,y_test=train_test_split(iris_X,iris_y,test_size=0.3)
 
       knn=KNeighborsClassifier()
       knn.fit(X_train,y_train)
 
       print(knn.predict(X_test))
 
       return y_test