Skip to content

Capstone project for Udacity Machine Learning Nanodegree

Notifications You must be signed in to change notification settings

jkielbaey/mlnd-capstone

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 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.


Releases

No releases published

Packages

No packages published