机器学习模型建立
逻辑回归分类
SVM分类
决策树分类
逻辑回归分类
from sklearn import linear_model
import pandas as pd
from sklearn.model_selection import train_test_split
df=pd.read_excel(‘data.xlsx’)
X=df[‘trans_failrate_last_1_month’]
y=df[‘status’]
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.3, random_state=42)
model=linear_model.LogisticRegression()
model.fit(X_train,y_train)
y_predict=model.predict(y_test)
sklearn.metrics.precision_score(y_test,y_predict)
SVM分类
from sklearn.svm import SVC
import pandas as pd
from sklearn.model_selection import train_test_split
df=pd.read_excel(‘data.xlsx’)
X=df[‘trans_failrate_last_1_month’]
y=df[‘status’]
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.3, random_state=42)
model = SVC()
model.fit(X_train,y_train)
y_predict=model.predict(y_test)
sklearn.metrics.precision_score(y_test,y_predict)
决策树分类
from sklearn.tree import DecisionTreeClassifier
import pandas as pd
from sklearn.model_selection import train_test_split
df=pd.read_excel(‘data.xlsx’)
X=df[‘trans_failrate_last_1_month’]
y=df[‘status’]
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.3, random_state=42)
model=DecisionTreeClassifier()
model.fit(X_train,y_train)
y_predict=model.predict(y_test)
sklearn.metrics.precision_score(y_test,y_predict)