|
|
| import pandas as pd
|
| from sklearn.model_selection import train_test_split
|
| from sklearn.ensemble import RandomForestClassifier
|
| from sklearn.metrics import accuracy_score
|
| import pickle
|
|
|
|
|
| df = pd.read_csv("mobile_prices.csv")
|
|
|
|
|
| X = df.drop("price_range", axis=1)
|
| y = df["price_range"]
|
|
|
|
|
| X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
|
|
|
|
|
| model = RandomForestClassifier()
|
| model.fit(X_train, y_train)
|
|
|
|
|
| y_pred = model.predict(X_test)
|
| print("Accuracy:", accuracy_score(y_test, y_pred))
|
|
|
|
|
| with open("model.pkl", "wb") as f:
|
| pickle.dump(model, f)
|
|
|