from sklearn.metrics import roc_curve,auc,confusion_matrix import pandas as pd import statsmodels.api as sm import matplotlib.pyplot as plt import numpy as np #库 w=pd.read_csv("D:/pythonProject/venv/Trans.csv") w['intercept']=1.0 x_c=w.columns[[4,0,1,2]];y_c=w.columns[3] X=w[x_c];y=w[y_c] X=np.array(X);y=np.array(y) zid=np.ones(len(y));zid[:int(len(y)*0.2)]=0 x_x=X[zid==1,:];y_x=y[zid==1] x_t=X[zid==0,:];y_t=y[zid==0] result=sm.Logit(y_x,x_x).fit print(result.summary)
时间: 2024-03-01 17:49:11 浏览: 109
RandomForest_sklearn.zip_sklearn_sklearn RF_southern9qq_随机森林
看起来您正在使用Python编写一个逻辑回归模型。在这段代码中,您首先导入所需的库,然后读取一个名为"Trans.csv"的文件,并将其存储在名为"w"的Pandas数据框中。接下来,您创建了一个名为"intercept"的新列,并将其值设置为1.0。然后,您选择了一些列作为自变量和一个列作为因变量,并将它们存储在名为"X"和"y"的NumPy数组中。接下来,您创建了一个名为"zid"的数组,并将其所有元素设置为1.0。然后,您将前20%的元素设置为0,以将数据集分成训练集和测试集。接下来,您将训练集和测试集分别存储在名为"x_x"和"x_t"的NumPy数组中,并将相应的标签存储在名为"y_x"和"y_t"的数组中。最后,您使用statsmodels.api库中的Logit函数来拟合逻辑回归模型,并将结果存储在名为"result"的变量中。您还打印了模型的摘要。
阅读全文