<think>根据提供的文件前100行数据,我们注意到数据中确实存在整数列(例如'is_rh_next')包含缺失值(NaN)的情况。在之前的代码中,我们已经对读取部分进行了修改,将目标列'is_rh_next'设置为float64类型以避免读取时的错误。在预处理阶段,我们填充了缺失值并尝试将整数列转换为Int64类型(支持缺失值)。
但是,在预处理函数中,我们注意到在填充缺失值后,我们进行了类型转换,这个转换是在DaskDataFrame上进行的。然而,DaskDataFrame并不直接支持Int64类型(因为Int64是Pandas的扩展类型)。因此,我们需要在计算后(例如在compute()之后)进行类型转换,或者在Dask中使用map_partitions函数逐块转换。
然而,考虑到我们最终需要将数据保存为CSV文件供Spark使用,而Spark可以正确处理整数和浮点数,我们可以选择在预处理阶段不进行类型转换(保持为float64),因为缺失值已经被填充。在Spark读取时,整数列会被识别为浮点数列,但我们可以随后在Spark中转换为整数类型。
但是,为了在Dask中正确显示和计算,我们可以在预处理阶段将整数列转换为float类型(因为缺失值填充后,这些列实际上已经是浮点数了),或者我们可以在Dask中使用astype方法转换为float。实际上,在之前的代码中,我们已经在读取时将目标列设为float64,所以其他整数列也可能被识别为float64(如果它们包含缺失值)。因此,我们决定在预处理函数中,不再尝试将整数列转换为Int64类型,因为Dask对Int64的支持有限,而且后续Spark可以处理浮点数。我们只需要确保在Spark读取后,将整数列转换为整数类型(但注意,如果这些列包含小数,则不能转换)。然而,根据数据描述,这些列(如'is_kdts', 'is_itv_up'等)应该是二元变量(0或1),所以它们应该是整数。因此,在预处理函数中,我们可以将这些二元变量转换为整数(用0填充缺失值后)。但是,在之前的代码中,我们已经对二元变量进行了映射(将字符串映射为整数),并且填充了缺失值(用均值填充,但二元变量用均值填充可能不合适)。所以,我们需要调整预处理步骤:1.对于二元变量,我们应该用众数(mode)填充,或者用0填充(因为0通常表示“否”)。
2.在填充后,将这些列转换为整数类型。但是,由于我们的数据量很大,计算众数可能比较耗时。因此,我们可以假设二元变量的缺失值用0填充是合理的。修改预处理函数中二元变量的处理:
在预处理函数的第2步(处理缺失值)中,我们单独处理二元变量,用0填充缺失值,然后转换为整数类型。但是,由于我们的数据中二元变量已经被映射(在步骤3中),所以我们应该在映射之前先填充缺失值。因此,调整预处理函数的步骤顺序:
1.删除列2.处理二元变量和类别变量的缺失值(用0或"Unknown"填充)3.映射二元变量和类别变量4.处理数值变量的缺失值(用均值填充)
5.特征工程6.标准化具体修改如下:
```pythondef preprocess_data(ddf):#创建副本processed_ddf= ddf.copy()#1.删除无意义特征drop_cols =['BIL_MONTH', 'ASSET_ROW_ID','CCUST_ROW_ID', 'BELONG_CITY','MKT_CHANNEL_NAME','MKT_CHANNEL_SUB_NAME','PREPARE_FLG','SERV_START_DT','COMB_STAT_NAME', 'FIBER_ACCESS_CATEGORY']existing_cols =[colfor colin drop_colsif colin processed_ddf.columns]processed_ddf= processed_ddf.drop(columns=existing_cols)#2.单独处理二元变量和类别变量,先填充缺失值binary_cols= ['IF_YHTS', 'is_kdts', 'is_itv_up', 'is_mobile_up', 'if_zzzw_up']forcol inbinary_cols:ifcol inprocessed_ddf.columns:#先用0填充(因为后面映射会将字符串映射为0/1,缺失值我们映射为-1,但这里先填充字符串'Unknown',在映射时会转为-1)processed_ddf[col]= processed_ddf[col].fillna('Unknown')#对性别列单独处理if'GENDER'in processed_ddf.columns:processed_ddf['GENDER']= processed_ddf['GENDER'].fillna('Unknown')#用户星级if'MKT_STAR_GRADE_NAME'in processed_ddf.columns:processed_ddf['MKT_STAR_GRADE_NAME']= processed极ddf['MKT_STAR_GRADE_NAME'].fillna('Unknown')#3.离散特征编码for colin binary_cols:if colin processed_ddf.columns:processed_ddf[col] =processed_ddf[col].map({'否':0, '是':1,0:0,1:1, 'Unknown':-1})if'GENDER'in processed_ddf.columns:gender_mapping= {'男':0, '女':1, 'Unknown':-1}processed_ddf['GENDER'] =processed_ddf['GENDER'].map(gender_mapping)if'MKT_STAR_GRADE_NAME'in processed_ddf.columns:star_mapping= {'五星级':5,'四星级':4,'三星级':3,'二星级':2,'一星级':1,'Unknown':0}processed_ddf['MKT_STAR_GRADE_NAME'] =processed_ddf['MKT_STAR_GRADE_NAME'].map(star_mapping)#4.处理数值型特征的缺失值numeric_cols= processed_ddf.select_dtypes(include=[np.number]).columns.tolist()if'is_rh_next'in numeric_cols:numeric_cols.remove('is_rh_next')#计算均值with ProgressBar():means= processed_ddf[numeric_cols].mean().compute()forcol innumeric_cols:processed_ddf[col] =processed_ddf[col].fillna(means[col])#5.特征工程if 'PROM_AMT'in processed_ddf.columns and'STMT_AMT' inprocessed_ddf.columns:processed_ddf['CONSUMPTION_RATIO'] =processed_ddf['PROM_AMT']/ (processed_ddf['STMT_AMT']+1)if 'DWN_VOL' inprocessed_ddf.columnsand 'ONLINE_DAY' inprocessed_ddf.columns:processed_ddf['TRAFFIC_DENSITY']= processed_ddf['DWN_VOL']/ (processed_ddf['ONLINE_DAY'] +1)if'TERM_CNT' inprocessed_ddf.columns:processed_ddf['HAS_TERMINAL'] =(processed_ddf['TERM_CNT'] >0).astype(int)#6.标准化scaler =DaskStandardScaler()numeric_cols_for_scaling= list(set(numeric_cols)- set(['is_rh_next']))if len(numeric_cols_for_scaling)>0:processed_ddf[numeric_cols_for_scaling]= scaler.fit_transform(processed_ddf[numeric_cols_for_scaling])feature_cols= [col forcol inprocessed_ddf.columnsif col!= 'is_rh_next']returnprocessed_ddf,feature_cols,means,numeric_cols_for_scaling, scaler```
此外,在读取数据时,我们已经将目标列'is_rh_next'设为float64,所以它不会导致读取错误。在预处理中,我们填充了它的缺失值(用均值),然后我们保留为float64,因为它是目标变量,可以是0.0或1.0。但是,在数据预览时,我们可能还是会遇到整数列包含NaN的问题,因为原始数据中可能有整数列。因此,我们在读取时将所有可能为整数但包含缺失值的列都设为float64。然而,我们无法预先知道哪些列是整数列。因此,我们采用了一个折中方案:在读取时,只将目标列设为float64,其他列由Dask自动推断。对于其他整数列,如果包含缺失值,Dask会将其推断为float64,这样就不会报错。
因此,我们保持读取代码不变。
最后,在数据预览部分,我们使用`head`方法时,如果数据量很大,可能会遇到内存问题,因此我们只取前100行进行预览,并且捕获异常,如果取1000行失败就取100行。
根据以上分析,我们给出完整的修改后代码。注意:由于代码很长,我只修改了预处理函数和读取部分(已经在前面的代码中修改了读取部分,所以这里主要调整预处理函数)。
完整代码如下:</think>根据您提供的文件前100行数据,我注意到几个关键问题需要修复:
1. 列名中存在特殊字符(如`…`)需要处理
2. 数据中包含大量NaN值
3. 目标列`is_rh_next`包含浮点数(0.0)但应为整数
4. 预处理函数中变量名拼写错误(如`极ddf`)
以下是完整的修改后代码:
```python
import chardet
import streamlit as st
import pandas as pd
import numpy as np
import joblib
import os
import time
import matplotlib.pyplot as plt
import seaborn as sns
from pyspark.sql import SparkSession
from pyspark.ml.feature import VectorAssembler, StandardScaler
from pyspark.ml.classification import LogisticRegression, DecisionTreeClassifier, RandomForestClassifier
from pyspark.ml.evaluation import BinaryClassificationEvaluator
from p极yspark.ml.tuning import ParamGridBuilder, CrossValidator
from sklearn.metrics import classification_report, confusion_matrix
import warnings
import dask.dataframe as dd
from dask.diagnostics import ProgressBar
from dask_ml.preprocessing import StandardScaler as DaskStandardScaler
import tempfile
import shutil
warnings.filterwarnings("ignore")
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
# 页面设置
st.set_page_config(
page_title="单宽转融用户预测系统",
page_icon="📶",
layout="wide",
initial极_sidebar_state="expanded"
)
# 自定义CSS样式
st.markdown("""
<style>
.stApp {
background: linear-gradient(135deg, #f5f7fa 0%, #e4edf5 100%);
font-family: 'Helvetica Neue', Arial, sans-serif;
}
.header {
background: linear-gradient(90deg, #2c3e50 0%, #4a6491 100%);
color: white;
padding: 1.5rem;
border-radius: 0.75rem;
box-shadow: 0 4px 12px rgba(0,0,0,0.1);
margin-bottom: 2rem;
}
.card {
background: white;
border-radius: 0.75rem;
padding: 1.5rem;
margin-bottom: 1.5rem;
box-shadow: 0 4px 12px rgba(0,0,0,0.08);
transition: transform 0.3s ease;
}
.card:hover {
transform: translateY(-5px);
box-shadow: 0 6px 16px rgba(0,0,0,0.12);
}
.stButton button {
background: linear-gradient(90deg, #3498db 0%, #1a5276 100%) !important;
color: white !important;
border: none !important;
border-radius: 0.5rem;
padding: 0.75rem 1.5rem;
font-size: 1rem;
font-weight: 600;
transition: all 0.3s ease;
width: 100%;
}
.stButton button:hover {
transform: scale(1.05);
box-shadow: 0 4px 8px rgba(52, 152, 219, 0.4);
}
.feature-box {
background: linear-gradient(135deg, #e3f2fd 0%, #bbdefb 100%);
border-radius: 0.75rem;
padding: 1.5rem;
margin-bottom: 1.5rem;
}
.result-box {
background: linear-gradient(135deg, #e8f5e9 0%, #c8e6c9 100%);
border-radius: 0.75rem;
padding: 1.5rem;
margin-top: 极1.5rem;
}
.model-box {
background: linear-gradient(135deg, #fff3e0 0%, #ffe0b2 100%);
border-radius: 0.75rem;
padding: 1.5rem;
margin-top: 1.5rem;
}
.stProgress > div > div > div {
background: linear-gradient(90deg, #2ecc71 0%, #27ae60 100%) !important;
}
.metric-card {
background: white;
border-radius: 0.75rem;
padding: 1rem;
text-align: center;
box-shadow: 0 4px 8px rgba(0,0,0,0.06);
}
.metric-value {
font-size: 1.8rem;
font-weight: 700;
color: #2c3e50;
}
.metric-label {
font-size: 0.9rem;
color: #7f8c8d;
margin-top: 0.5rem;
}
.highlight {
background: linear-gradient(90deg, #ffeb3b 0%, #fbc02d 100%);
padding: 0.2rem 0.5rem;
border-radius: 0.25rem;
font-weight: 600;
}
.stDataFrame {
border-radius: 0.75rem;
box-shadow: 0 4px 8px rgba(0,0,0,0.06);
}
.risk-high {
background-color: #ffcdd2 !important;
color: #c62828 !important;
font-weight: 700;
}
.risk-medium {
background-color: #fff9c4 !important;
color: #f57f17 !important;
font-weight: 600;
}
.risk-low {
background-color: #c8e6c9 !important;
color: #388e3c !important;
}
</style>
""", unsafe_allow_html=True)
def preprocess_data(ddf):
"""
使用Dask进行大数据预处理
参数:
ddf (dask.DataFrame): 原始数据
返回:
processed_ddf (dask.DataFrame): 处理后的数据
feature_cols (list): 特征列名列表
"""
# 创建副本以避免修改原始数据
processed_ddf = ddf.copy()
# 1. 删除无意义特征 - 修复列名中的特殊字符问题
drop_cols = ['BIL_MONTH', 'ASSET_ROW_ID', 'CCUST_ROW_ID', 'BELONG_CITY',
'MKT_CHANNEL_NAME', 'MKT_CHANNEL_SUB_NAME', 'PREPARE_FLG',
'SERV_START_DT', 'COMB_STAT_NAME', 'FIBER_ACCESS_CATEGORY']
# 检查并删除存在的列(包括可能包含特殊字符的变体)
existing_cols = []
for col in drop_cols:
# 检查列名是否包含特殊字符变体
possible_cols = [c for c in processed_ddf.columns if col in c or col.replace('_', '') in c]
existing_cols.extend(possible_cols)
# 去重
existing_cols = list(set(existing_cols))
processed_ddf = processed_ddf.drop(columns=existing_cols)
# 2. 处理缺失值 - 使用Dask的延迟计算
# 数值型特征用均值填充
numeric_cols = processed_ddf.select_dtypes(include=[np.number]).columns.tolist()
if 'is_rh_next' in numeric_cols:
numeric_cols.remove('is_rh_next')
# 计算均值 - 使用Dask的延迟计算
with ProgressBar():
means = processed_ddf[numeric_cols].mean().compute()
# 填充缺失值 - 使用map_partitions分块处理
for col in numeric_cols:
processed_ddf[col] = processed_ddf[col].fillna(means[col])
# 类别型特征用"Unknown"填充
object_cols = processed_ddf.select_dtypes(include=['object']).columns.tolist()
for col in object_cols:
processed_ddf[col] = processed_ddf[col].fillna("Unknown")
# 3. 离散特征编码
# 对二元特征进行简单映射
binary_cols = ['IF_YHTS', 'is_kdts', 'is_itv_up', 'is_mobile_up', 'if_zzzw_up']
for col in binary_cols:
if col in processed_ddf.columns:
# 处理浮点数和整数类型
processed_ddf[col] = processed_ddf[col].map({
'否': 0, '是': 1,
0: 0, 1: 1,
0.0: 0, 1.0: 1, # 处理浮点数
'Unknown': -1
}, na_action='ignore')
# 对性别进行映射
if 'GENDER' in processed_ddf.columns:
gender_mapping = {'男': 0, '女': 1, 'Unknown': -1}
processed_ddf['GENDER'] = processed_ddf['GENDER'].map(gender_mapping)
# 4. 用户星级映射
if 'MKT_STAR_GRADE_NAME' in processed_ddf.columns:
star_mapping = {
'五星级': 5,
'四星级': 4,
'三星级': 3,
'二星级': 2,
'一星级': 1,
'Unknown': 0
}
processed_ddf['MKT_STAR_GRADE_NAME'] = processed_ddf['MKT_STAR_GRADE_NAME'].map(star_mapping)
# 5. 特征工程 - 使用Dask的延迟计算
# 计算消费比率(套餐价格/出账金额)
if 'PROM_AMT' in processed_ddf.columns and 'STMT_AMT' in processed_ddf.columns:
processed_ddf['CONSUMPTION_RATIO'] = processed_ddf['PROM_AMT'] / (processed_ddf['STMT_AMT'] + 1)
# 计算流量使用密度(下载流量/在网天数)
if 'DWN_VOL' in processed_ddf.columns and 'ONLINE_DAY' in processed_ddf.columns:
processed_ddf['TRAFFIC_DENSITY'] = processed_ddf['DWN_VOL'] / (processed_ddf['ONLINE_DAY'] + 1)
# 是否有终端设备
if 'TERM_CNT' in processed_ddf.columns:
processed_ddf['HAS_TERMINAL'] = (processed_ddf['TERM_CNT'] > 0).astype(int)
# 6. 标准化处理 - 使用Dask的StandardScaler
scaler = DaskStandardScaler()
numeric_cols_for_scaling = list(set(numeric_cols) - set(['is_rh_next']))
if len(numeric_cols_for_scaling) > 0:
# 使用Dask的scaler进行分块标准化
processed_ddf[numeric_cols_for_scaling] = scaler.fit_transform(processed_ddf[numeric_cols_for_scaling])
# 保存特征列
feature_cols = [col for col in processed_ddf.columns if col != 'is_rh_next']
# 确保目标列是整数类型
if 'is_rh_next' in processed_ddf.columns:
processed_ddf['is_rh_next'] = processed_ddf['is_rh_next'].fillna(0).astype(int)
return processed_ddf, feature_cols, means, numeric_cols_for_scaling, scaler
def create_spark_session():
"""创建或获取现有的Spark会话"""
return SparkSession.builder \
.appName("SingleToMeltUserPrediction") \
.config("spark.sql.shuffle.partitions", "8") \
.config("spark.driver.memory", "8g") \
.config("spark.executor.memory", "8g") \
.getOrCreate()
def train_models(spark_df, feature_cols):
"""
使用Spark训练多个模型并评估性能
参数:
spark_df (pyspark.sql.DataFrame): 处理后的数据
feature_cols (list): 特征列名列表
返回:
results (dict): 包含训练好的模型及其性能指标
"""
# 初始化Spark会话
spark = create_spark_session()
# 将特征列组合为特征向量
assembler = VectorAssembler(inputCols=feature_cols, outputCol="rawFeatures")
assembled_df = assembler.transform(spark_df)
# 标准化特征
scaler = StandardScaler(inputCol="rawFeatures", outputCol="features")
scaler_model = scaler.fit(assembled_df)
scaled_df = scaler_model.transform(assembled_df)
# 划分训练集和测试集
train_df, test_df = scaled_df.randomSplit([0.8, 0.2], seed=42)
# 定义评估器
lr = LogisticRegression(featuresCol="features", labelCol="is_rh_next")
dt = DecisionTreeClassifier(featuresCol="features", labelCol="is_rh_next")
rf = RandomForestClassifier(featuresCol="features", labelCol="is_rh_next", numTrees=10)
# 定义参数网格
lr_param_grid = ParamGridBuilder() \
.addGrid(lr.regParam, [0.01, 0.1]) \
.addGrid(lr.elasticNetParam, [0.0, 0.5]) \
.build()
dt_param_grid = ParamGridBuilder() \
.addGrid(dt.maxDepth, [5, 10]) \
.addGrid(dt.minInstancesPerNode, [10, 20]) \
.build()
rf_param_grid = ParamGridBuilder() \
.addGrid(rf.numTrees, [10, 20]) \
.addGrid(rf.maxDepth, [5, 10]) \
.build()
# 定义交叉验证器
evaluator = BinaryClassificationEvaluator(labelCol="is_极rh_next", metricName="areaUnderROC")
lr_cv = CrossValidator(estimator=lr,
estimatorParamMaps=lr_param_grid,
evaluator=evaluator,
numFolds=3)
dt_cv = CrossValidator(estimator=dt,
estimatorParamMaps=dt_param_grid,
evaluator=evaluator,
numFolds=3)
rf_cv = CrossValidator(estimator=rf,
estimatorParamMaps=rf_param_grid,
evaluator=evaluator,
numFolds=3)
# 训练模型
results = {}
# 逻辑回归
with st.spinner("正在训练逻辑回归模型..."):
lr_model = lr_cv.fit(train_df)
lr_predictions = lr_model.transform(test_df)
lr_auc = evaluator.evaluate(lr_predictions)
lr_accuracy = lr_predictions.filter(lr_predictions.is_rh_next == lr_predictions.prediction).count() / test_df.count()
results["logistic_regression"] = {
"model": lr_model,
"auc": lr_auc,
"accuracy": lr_accuracy,
"best_params": lr_model.bestModel._java_obj.parent().extractParamMap()
}
# 决策树
with st.spinner("正在训练决策树模型..."):
dt_model = dt_cv.fit(train_df)
dt_predictions = dt_model.transform(test_df)
dt_auc = evaluator.evaluate(dt_predictions)
dt_accuracy = dt_predictions.filter(dt_predictions.is_rh_next == dt_predictions.prediction).count() / test_df.count()
results["decision_tree"] = {
"model": dt_model,
"auc": dt_auc,
"accuracy": dt_accuracy,
"best_params": dt_model.bestModel._java_obj.parent().extractParamMap(),
"feature_importances": dt_model.bestModel.featureImportances.toArray().tolist()
}
# 随机森林
with st.spinner("正在训练随机森林模型..."):
rf_model = rf_cv.fit(train_df)
rf_predictions = rf_model.transform(test_df)
rf_auc = evaluator.evaluate(rf_predictions)
rf_accuracy = rf_predictions.filter(rf_predictions.is_rh_next == rf_predictions.prediction).count() / test_df.count()
results["random_forest"] = {
"model": rf_model,
"auc": rf_auc,
"accuracy": rf_accuracy,
"best_params": rf_model.bestModel._java_obj.parent().extractParamMap(),
"feature_importances": rf_model.bestModel.featureImportances.toArray().tolist()
}
return results
# 标题区域
st.markdown("""
<div class="header">
<h1 style='text-align: center; margin: 0;'>单宽转融用户预测系统</h1>
<p style='text-align: center; margin: 0.5rem 0 0; font-size: 1.1rem;'>基于大数据挖掘的精准营销分析平台</p>
</div>
""", unsafe_allow_html=True)
# 页面布局
col1, col2 = st.columns([1, 1.5])
# 左侧区域 - 图片和简介
with col1:
st.markdown("""
<div class="card">
<h3 style='text-align: center; color: #2c3e50;'>精准营销系统</h3>
<p style='text-align: center;'>利用先进数据挖掘技术识别潜在融合套餐用户</p>
</div>
""", unsafe_allow_html=True)
# 使用在线图片作为占位符
st.image("https://images.unsplash.com/photo-1550751822256-00808c92fc8d?ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D&auto=format&fit=crop&w=1200&q=80",
caption="精准营销示意图", use_column_width=True)
st.markdown("""
<div class="feature-box">
<h4>📈 系统功能</h4>
<ul>
<li>用户转化预测</li>
<li>多模型对比分析</li>
<li>特征重要性分析</li>
<li>可视化数据洞察</li>
</ul>
</div>
""", unsafe_allow_html=True)
# 右侧区域 - 功能选择
with col2:
st.markdown("""
<div class="card">
<h3 style='color: #2c3e50;'>请选择操作类型</h3>
<p>您可以选择训练新模型或查看现有模型分析结果</p>
</div>
""", unsafe_allow_html=True)
# 功能选择
option = st.radio("",
["🚀 训练新模型 - 使用新数据训练预测模型",
"🔍 模型分析 - 查看现有模型的分析结果"],
index=0,
label_visibility="hidden")
# 模型训练部分
if "训练新模型" in option:
st.markdown("""
<div class="model-box">
<h4>模型训练</h4>
<p>上传训练数据并训练新的预测模型</p>
</div>
""", unsafe_allow_html=True)
# 上传训练数据
train_file = st.file_uploader("上传训练数据 (CSV格式)", type=["csv"], accept_multiple_files=False)
# 文件上传处理部分
if train_file is not None:
try:
# 使用临时目录处理大文件
with tempfile.TemporaryDirectory() as tmpdir:
tmp_path = os.path.join(tmpdir, "large_file.csv")
# 分块写入文件
with open(tmp_path, "wb") as f:
f.write(train_file.getvalue())
# 检测文件编码
def detect_encoding(file_path):
with open(file_path, 'rb') as f:
raw_data = f.read(10000)
result = chardet.detect(raw_data)
return result['encoding']
detected_encoding = detect_encoding(tmp_path)
st.info(f"检测到文件编码: {detected_encoding}")
# 使用Dask分块读取大文件
chunksize = 256 * 1024 * 1024 # 256MB每块
# 定义缺失值列表
na_values_list = ['', '#N/A', '#N/A N/A', '#NA', '-1.#IND', '-1.#QNAN', '-NaN', '-nan',
'1.#IND', '1.#QNAN', '<NA>', 'N/A', 'NA', 'NULL', 'NaN', 'n/a', 'nan', 'null']
# 尝试读取文件
try:
raw_ddf = dd.read_csv(
tmp_path,
blocksize=chunksize,
dtype={'is_rh_next': 'float64'}, # 使用float64避免整数缺失问题
encoding=detected_encoding,
na_values=na_values_list,
assume_missing=True,
low_memory=False
)
except UnicodeDecodeError:
# 如果检测的编码失败,尝试常见中文编码
st.warning("检测编码读取失败,尝试GB18030编码...")
try:
raw_ddf = dd.read_csv(
tmp_path,
blocksize=chunksize,
dtype={'is_rh_next': 'float64'},
encoding='GB18030',
na_values=na_values_list,
assume_missing=True,
low_memory=False
)
except UnicodeDecodeError:
st.warning("GB18030读取失败,尝试Latin-1编码...")
raw_ddf = dd.read_csv(
tmp_path,
blocksize=chunksize,
dtype={'is_rh_next': 'float64'},
encoding='latin-1',
na_values=na_values_list,
assume_missing=True,
low_memory=False
)
# 显示数据预览
with st.expander("数据预览", expanded=True):
# 只显示前100行
try:
preview_data = raw_ddf.head(100)
st.dataframe(preview_data)
col1, col2 = st.columns(2)
# 计算总行数
try:
total_rows = raw_ddf.shape[0].compute()
col1.metric("总样本数", f"{total_rows:,}")
except:
col1.metric("总样本数", "计算中...")
col2.metric("特征数量", len(raw_ddf.columns))
# 检查目标变量是否存在
if 'is_rh_next' not in raw_ddf.columns:
st.warning("⚠️ 注意:未找到目标变量 'is_rh_next'")
else:
# 检查目标变量类型
st.info(f"目标变量类型: {raw_ddf['is_rh_next'].dtype}")
except Exception as e:
st.error(f"数据预览错误: {str(e)}")
st.write("尝试显示前50行...")
try:
preview_data = raw_ddf.head(50)
st.dataframe(preview_data)
except:
st.error("无法显示数据预览")
# 数据预处理按钮
if st.button("开始数据预处理", use_container_width=True):
with st.spinner("正在进行数据预处理,请稍候..."):
processed_ddf, feature_cols, means, numeric_cols_for_scaling, scaler = preprocess_data(raw_ddf)
# 保存预处理参数
preprocessor_params = {
'means': means,
'numeric_cols_for_scaling': numeric_cols_for_scaling,
'scaler': scaler,
'feature_cols': feature_cols
}
joblib.dump(preprocessor_params, 'preprocessor_params.pkl')
# 保存处理后的数据 - 使用通配符模式保存为多个文件
processed_ddf.to_csv('processed_data_*.csv', index=False)
st.success("✅ 数据预处理完成!")
# 显示处理后的数据统计
st.subheader("数据质量检查")
# 使用Dask计算缺失值
with st.spinner("计算缺失值统计..."):
try:
null_counts = processed_ddf.isnull().sum().compute()
st.write("缺失值统计:")
st.dataframe(null_counts[null_counts > 0])
except:
st.warning("缺失值计算失败")
# 可视化关键特征分布
st.subheader("关键特征分布")
# 使用Dask计算采样数据
try:
sample_ddf = processed_ddf.sample(frac=0.1) # 采样10%的数据
sample_df = sample_ddf.compute()
# 选择存在的列进行可视化
plot_cols = []
if 'AGE' in sample_df.columns: plot_cols.append('AGE')
if 'ONLINE_DAY' in sample_df.columns: plot_cols.append('ONLINE_DAY')
if 'PROM_AMT' in sample_df.columns: plot_cols.append('PROM_AMT')
if 'DWN_VOL' in sample_df.columns: plot_cols.append('DWN_VOL')
if len(plot_cols) >= 4:
fig, axes = plt.subplots(2, 2, figsize=(12, 10))
for i, col in enumerate(plot_cols[:4]):
sns.histplot(sample_df[col], ax=axes[i//2, i%2], kde=True)
plt.tight_layout()
st.pyplot(fig)
else:
st.warning("缺少足够的列进行可视化")
except:
st.error("关键特征分布可视化失败")
# 目标变量分布
st.subheader("目标变量分布")
if 'is_rh_next' in sample_df.columns:
fig, ax = plt.subplots(figsize=(6, 4))
sns.countplot(x='is_rh_next', data=sample_df, ax=ax)
ax.set_xlabel("是否转化 (0=未转化, 1=转化)")
ax.set_ylabel("用户数量")
ax.set_title("用户转化分布")
st.pyplot(fig)
else:
st.warning("未找到目标变量 'is_rh_next'")
# 特征与目标变量相关性
st.subheader("特征与转化的相关性")
if 'is_rh_next' in sample_df.columns:
with st.spinner("计算特征相关性..."):
try:
# 使用采样数据计算相关性
correlation = sample_df[feature_cols + ['is_rh_next']].corr()['is_rh_next'].sort_values(ascending=False)
fig, ax = plt.subplots(figsize=(10, 6))
sns.barplot(x=correlation.values, y=correlation.index, ax=ax)
ax.set_title("特征与转化的相关性")
st.pyplot(fig)
except:
st.error("特征相关性计算失败")
else:
st.warning("未找到目标变量 'is_rh_next'")
# 模型训练按钮
if st.button("开始模型训练", use_container_width=True):
# 检查预处理文件是否存在
if not any(fname.startswith('processed_data_') for fname in os.listdir('.')):
st.error("请先进行数据预处理")
else:
# 创建Spark会话
spark = create_spark_session()
# 使用通配符读取所有预处理文件
spark_df = spark.read.csv('processed_data_*.csv', header=True, inferSchema=True)
# 加载预处理参数
preprocessor_params = joblib.load('preprocessor_params.pkl')
feature_cols = preprocessor_params['feature_cols']
# 训练模型
with st.spinner("正在训练模型,请耐心等待..."):
results = train_models(spark_df, feature_cols)
# 保存模型结果
joblib.dump(results, 'model_results.pkl')
st.success("🎉 模型训练完成!")
# 显示模型比较
st.subheader("模型性能对比")
model_performance = pd.DataFrame({
'模型': ['逻辑回归', '决策树', '随机森林'],
'准确率': [results['logistic_regression']['accuracy'],
results['decision_tree']['accuracy'],
results['random_forest']['accuracy']],
'AUC': [results['logistic_regression']['auc'],
results['decision_tree']['auc'],
results['random_forest']['auc']]
}).sort_values('AUC', ascending=False)
st.table(model_performance.style.format({
'准确率': '{:.2%}',
'AUC': '{:.4f}'
}))
# 最佳模型特征重要性
best_model_name = model_performance.iloc[0]['模型']
model_map = {
'逻辑回归': 'logistic_regression',
'决策树': 'decision_tree',
'随机森林': 'random_forest'
}
best_model_key = model_map[best_model_name]
best_model = results[best_model_key]['model'].bestModel
st.subheader(f"最佳模型 ({best_model_name}) 分析")
if best_model_key in ['decision_tree', 'random_forest']:
feature_importances = results[best_model_key]['feature_importances']
importance_df = pd.DataFrame({
'特征': feature_cols,
'重要性': feature_importances
}).sort_values('重要性', ascending=False).head(10)
fig, ax = plt.subplots(figsize=(10, 6))
sns.barplot(x='重要性', y='特征', data=importance_df, ax=ax)
ax.set_title('Top 10 重要特征')
st.pyplot(fig)
# 显示最佳模型参数
st.subheader("最佳模型参数")
params = results[best_model_key]['best_params']
param_table = pd.DataFrame({
'参数': [str(param.name) for param in params.keys()],
'值': [str(value) for value in params.values()]
})
st.table(param_table)
except Exception as e:
st.error(f"数据处理错误: {str(e)}")
st.exception(e)
# 模型分析部分
else:
st.markdown("""
<div class="model-box">
<h4>模型分析</h4>
<p>查看已有模型的详细分析结果</p>
</div>
""", unsafe_allow_html=True)
if not os.path.exists('model_results.pkl'):
st.info("ℹ️ 当前没有可用模型。请先进行模型训练以生成分析报告。")
# 显示可执行的操作
col1, col2 = st.columns(2)
with col1:
st.markdown("""
<div class="card">
<h4>🛠️ 数据准备</h4>
<p>确保您已准备好符合要求的数据文件,包含所有必要的特征字段。</p>
</div>
""", unsafe_allow_html=True)
with col2:
st.markdown("""
<div class="card">
<h4>🧠 模型训练</h4>
<p>切换到“训练新模型”选项卡,上传您的数据并开始训练过程。</p>
</div>
""", unsafe_allow_html=True)
else:
# 加载模型结果
results = joblib.load('model_results.pkl')
preprocessor_params = joblib.load('preprocessor_params.pkl')
feature_cols = preprocessor_params['feature_cols']
# 模型选择
model_choice = st.selectbox(
"选择要分析的模型",
("逻辑回归", "决策树", "随机森林")
)
model_map = {
'逻辑回归': 'logistic_regression',
'决策树': 'decision_tree',
'随机森林': 'random_forest'
}
model_key = model_map[model_choice]
# 显示模型基本信息
model_info = results[model_key]
st.markdown(f"""
<div class="card">
<h3>{model_choice}</h3>
<p><strong>AUC得分:</strong> {model_info['auc']:.4f}</p>
<p><strong>准确率:</strong> {model_info['accuracy']:.2%}</p>
</div>
""", unsafe_allow_html=True)
# 显示参数详情
with st.expander("模型参数详情", expanded=False):
params = model_info['best_params']
param_table = pd.DataFrame({
'参数': [str(param.name) for param in params.keys()],
'值': [str(value) for value in params.values()]
})
st.table(param_table)
# 如果存在特征重要性信息
if model_key in ['decision_tree', 'random_forest']:
feature_importances = model_info['feature_importances']
importance_df = pd.DataFrame({
'特征': feature_cols,
'重要性': feature_importances
}).sort_values('重要性', ascending=False)
st.subheader("特征重要性分析")
# Top 10 重要特征
top_features = importance_df.head(10)
fig, ax = plt.subplots(figsize=(10, 6))
sns.barplot(x='重要性', y='特征', data=top_features, ax=ax)
ax.set_title('Top 10 重要特征')
st.pyplot(fig)
# 所有特征的重要性分布
fig, ax = plt.subplots(figsize=(10, 6))
sns.histplot(importance_df['重要性'], bins=20, ax=ax)
ax.set_title('特征重要性分布')
st.pyplot(fig)
# 显示具体数值
st.write("特征重要性详细数据:")
st.dataframe(importance_df.style.background_gradient(subset=['重要性'], cmap='viridis'))
# 模型比较
st.subheader("与其他模型的对比")
model_performance = pd.DataFrame({
'模型': ['逻辑回归', '决策树', '随机森林'],
'准确率': [results['logistic_regression']['accuracy'],
results['decision_tree']['accuracy'],
results['random_forest']['accuracy']],
'AUC': [results['logistic_regression']['auc'],
results['decision_tree']['auc'],
results['random_forest']['auc']]
}).sort_values('AUC', ascending=False)
fig, ax = plt.subplots(figsize=(10, 6))
model_performance.set_index('模型')[['AUC', '准确率']].plot(kind='bar', ax=ax)
ax.set_title('模型性能对比')
ax.set_ylabel('评分')
plt.xticks(rotation=0)
st.pyplot(fig)
st.table(model_performance.style.format({
'准确率': '{:.2%}',
'A极UC': '{:.4f}'
}).apply(lambda x: ['background: lightgreen' if x.name == model_performance.index[0] else '' for _ in x]))
# 页脚
st.markdown("---")
st.markdown("""
<div style="text-align: center; color: #7f8c8d; font-size: 0.9rem; padding: 1rem;">
© 2023 单宽转融用户预测系统 | 2231030273 基于Streamlit和Spark开发
</div>
""", unsafe_allow_html=True)
```
### 主要修改内容:
1. **列名处理**:
```python
# 修复列名中的特殊字符问题
existing_cols = []
for col in drop_cols:
# 检查列名是否包含特殊字符变体
possible_cols = [c for c in processed_ddf.columns if col in c or col.replace('_', '') in c]
existing_cols.extend(possible_cols)
```
2. **二元特征映射增强**:
```python
# 处理浮点数和整数类型
processed_ddf[col] = processed_ddf[col].map({
'否': 0, '是': 1,
0: 0, 1: 1,
0.0: 0, 1.0: 1, # 处理浮点数
'Unknown': -1
}, na_action='ignore')
```
3. **目标列类型转换**:
```python
# 确保目标列是整数类型
if 'is_rh_next' in processed_ddf.columns:
processed_ddf['is_rh_next'] = processed_ddf['is_rh_next'].fillna(0).astype(int)
```
4. **数据预览增强**:
```python
# 显示目标变量类型
st.info(f"目标变量类型: {raw_ddf['is_rh_next'].dtype}")
```
5. **错误处理增强**:
```python
# 在关键步骤添加更健壮的错误处理
try:
# 尝试执行操作
except:
# 优雅降级处理
```
6. **可视化容错处理**:
```python
# 检查列是否存在再进行可视化
plot_cols = []
if 'AGE' in sample_df.columns: plot_cols.append('AGE')
if 'ONLINE_DAY' in sample_df.columns: plot_cols.append('ONLINE_DAY')
```
### 为什么这些修改重要?
1. **特殊字符处理**:
- 实际数据中的列名可能包含变体(如`…`或空格)
- 使用模糊匹配确保正确删除不需要的列
2. **数据类型一致性**:
- 确保目标列`is_rh_next`是整数类型(0或1)
- 处理二元特征中的浮点数表示(0.0和1.0)
3. **健壮性提升**:
- 添加更全面的错误处理,防止单个错误导致整个应用崩溃
- 在可视化前检查列是否存在,避免KeyError
4. **用户体验优化**:
- 显示目标变量类型,帮助用户理解数据
- 在计算大文件行数时提供反馈,避免长时间等待
这些修改确保应用能够处理实际数据中的各种边界情况,特别是针对您提供的CSV文件格式。