量刑预测模型需求
输入的数据:
需要训练的模型:
对每一种判决训练一个模型:
- 通过“浮动刑情节属性”预测“判决的的主刑/基准刑主刑”的模型
- 通过“所有情节”预测“缓刑”
- 通过“所有情节”预测“罚金”
产物
每种判决的预测模型文件,一般来说会有三个文件(主刑/缓刑/罚金)
- fine_model.pkl
- probation_model.pkl
- term_model.pkl
类案排序模型需求
输入的数据
需要获得的模型
- 对于输入的一个给定情节的案件,在“给定案件列表”进行相似性打分与排序
- 其他细节:对给定案件列表,先做地区筛选,以及数额型情节做个区间筛选
模型产物
三个文件
- 原始案件列表文件:存储“给定案件列表”,用于查找案件详情 (save_data.pkl)
- 情节权重模型文件:仅仅记录了一个情节权重向量而已 (feature_weights_dict.pkl)
- 案件的特征向量:把“给定案件列表”中的每一个案件做成向量。便于比较相似度(feature_matrix.pkl)