Как написать научную оценку в PyTorch - PullRequest
0 голосов
/ 02 мая 2020

Я разработал оценщик в Scikit-learn, но из-за проблем с производительностью (и скоростью, и использованием памяти) я думаю о том, чтобы заставить оценщик работать с использованием графического процессора.

Один способ, которым я могу придумать, чтобы сделать это написать оценщик в PyTorch (чтобы я мог использовать обработку GPU), а затем использовать Google Colab, чтобы использовать свои облачные графические процессоры и объем памяти.

Какой самый лучший способ написать оценку, которая уже совместимость с scikit-Learn в PyTorch?

Любые указатели или подсказки, указывающие в правильном направлении, будут действительно оценены. Большое спасибо заранее.

...