Блог

Я занимаюсь проектом машинного обучения, в котором мне нужно отображать прогнозы на веб-странице. Веб-страница построена с использованием Django. У меня есть функция предсказаний и веса модели, но как интегрировать функцию предсказаний, модель и веса в код Django и делать прогнозы.

Мой код предсказания

 def predicting(model, device, loader):
    model.eval()
    total_preds = torch.Tensor()
    total_labels = torch.Tensor()
    with torch.no_grad():
        for solute_graphs, solvent_graphs, solute_lens, solvent_lens in loader:
            outputs, i_map = model(
            [solute_graphs.to(device), solvent_graphs.to(device), torch.tensor(solute_lens).to(device),
             torch.tensor(solvent_lens).to(device)])   
            print(outputs)
            total_preds = torch.cat((total_preds, outputs.cpu()), 0)
    return total_preds.numpy().flatten()

 

Я сохранил веса в .tar файле, поэтому мне нужно запустить модель при загрузке весов для прогнозирования. Я понятия не имею, где хранить мою модель PyTorch и веса для вывода с помощью Django. пожалуйста, помогите.

Разверните классификатор глубокого обучения PyTorch в Heroku с помощью Django за 30 минут

https://www.youtube.com/watch?v=MLk2We1rJPs

Использование PyTorch внутри приложения Django:

https://stefanbschneider.github.io/blog/pytorch-django

Я надеюсь, что это поможет вам начать!!!