Skip to content

jkielbaey/mlnd-capstone

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Machine Learning Nanodegree - Capstone project

This repository contains my capstone project for Udacity Machine Learning Nanodegree.

In this project I addressed the problem of classification of high-energy particle collisions into either Signal or Background events in the context of detection of Higgs Boson.

For more details, see the proposal and final report

Technology stack

In my project I choose to use Amazon SageMaker for all aspects related to data exploration/visualization, training and deployment.

The Deep Neural Network is implemented using PyTorch.

As part of the project I also implemented a REST API to make predictions. The API is using Amazon API Gateway and AWS Lambda. For inference, the Lambda function calls the PyTorch model which was deployed as a SageMaker Endpoint.