CandyMK/iris_app
0
1import pandas as pd2import pickle3from sklearn.linear_model import LogisticRegression4from sklearn.model_selection import train_test_split5 6# memuat dataset Iris7from sklearn.datasets import load_iris8iris = load_iris()9df = pd.DataFrame(iris.data, columns=['sepal.length', 'sepal.width', 'petal.length', 'petal.width'])10df['target'] = iris.target11 12 13# split data14X = df.drop('target', axis=1)15y = df['target']16X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)17 18 19# latih model20model = LogisticRegression(max_iter=200)21model.fit(X_train, y_train)22 23 24# option 1: simpan model dengan PICKLE25with open('logreg_model.pkl', 'wb') as file:26 pickle.dump(model, file)27 28print("Model berhasil disimpan dalam format pickle")