CoolFace
Apppublic

Edohanifauzan/Edo_app

sourceHugging Faceotherupdated 4y agoView on Hugging Face
0likes
import.py27 linesDownload Raw Back to root
1import pandas as pd2raisin=pd.read_csv('https://raw.githubusercontent.com/EdoHanifauzan/data/Dataset/Raisin_Dataset.csv')3# Ordinal feature encoding4# (https://www.kaggle.com/datasets/shrutisaxena/raisin-dataset ")5df = raisin.copy()6target = 'Class'7 8target_mapper = {'Kecimen':0, 'Besni':1}9def target_encode(val):10    return target_mapper[val]11 12df['Class'] = df['Class'].apply(target_encode)13 14# Separating X and y15X = df.drop('Class', axis=1)16Y = df['Class']17 18# Build random forest model19 20from sklearn.naive_bayes import GaussianNB21model = GaussianNB()22model.fit(X, Y)23 24# Saving the model25import pickle26pickle.dump(model, open('modelNBC_Raisin.pkl', 'wb'))27