MST

星途 面试题库

面试题:Python多条件判断与复杂机器学习模型的融合

假设有一个基于决策树的分类模型,用于预测客户是否会购买产品(0表示不购买,1表示购买)。同时存在三个条件:客户年龄大于30岁、客户年收入大于50万、客户过往购买次数大于3次。若模型预测客户会购买产品且满足上述三个条件中的至少两个,输出'重点关注客户',否则输出'普通客户'。请用Python实现此逻辑,要求代码结构清晰,可运行。
50.1万 热度难度
编程语言Python

知识考点

AI 面试

面试题答案

一键面试
def classify_customer(age, income, purchase_count):
    condition1 = age > 30
    condition2 = income > 50
    condition3 = purchase_count > 3
    condition_count = sum([condition1, condition2, condition3])
    is_purchase = 1  # 这里假设模型预测客户会购买产品,实际应用中从模型获取预测值
    if is_purchase and condition_count >= 2:
        return '重点关注客户'
    else:
        return '普通客户'


# 示例调用
age = 35
income = 60
purchase_count = 4
print(classify_customer(age, income, purchase_count))