Python 大数据决策项目实训:识别运营商低价值用户
import pandas as pd\nfrom sklearn.model_selection import train_test_split\nfrom sklearn.preprocessing import StandardScaler\nfrom sklearn.ensemble import RandomForestClassifier\nfrom sklearn.metrics import classification_report\ndata = pd.read_csv('user_data.csv')\nprint(data.head())\nprint(data.info())\ndata = data.drop(['user_id', 'registration_date'], axis=1)\ndata = data.dropna()\nX = data.drop('target', axis=1)\ny = data['target']\nX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)\nscaler = StandardScaler()\nX_train = scaler.fit_transform(X_train)\nX_test = scaler.transform(X_test)\nmodel = RandomForestClassifier()\nmodel.fit(X_train, y_train)\ny_pred = model.predict(X_test)\nprint(classification_report(y_test, y_pred))
原文地址: https://www.cveoy.top/t/topic/pBK7 著作权归作者所有。请勿转载和采集!