daxia13 2022-07-29 19:16
浏览 59
已结题

catboost的gpu训练

请问catboost是不是有什么gpu版本?或者类似的,想用gpu加速catboost但是每次都变成cpu版本的,究竟该怎么实现catboost的gpu加速呢,除了代码里是不是还有别的什么要做的

model = CatBoostClassifier(
    iterations=10000,
    learning_rate=0.01,
    loss_function='Logloss',
    depth=10,
    task_type = 'gpu',
    random_seed = 42,
    od_type = "Iter",
    )


  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 8月6日
    • 创建了问题 7月29日