New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

First CPU implementation of HashedNets compression algorithm for DNNs #220

Closed
wants to merge 42 commits into
base: master
from
Commits
Jump to file or symbol
Failed to load files and symbols.
+1,372 −1,195
Diff settings

Always

Just for now

@@ -1,5 +1,5 @@
// @(#)root/tmva $Id$
// Author: Peter Speckmayer
// Authors: Peter Speckmayer, Aditya Sharma
/**********************************************************************************
* Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
@@ -10,7 +10,8 @@
* Description: *
* NeuralNetwork *
* *
* Authors (alphabetical): *
* Authors (alphabetical):
* Aditya Sharma <adisharma075@gmail.com> - CERN, Switzerland
* Peter Speckmayer <peter.speckmayer@gmx.at> - CERN, Switzerland *
* *
* Copyright (c) 2005-2015: *
@@ -70,12 +71,10 @@ namespace TMVA {
MethodDNN ( const TString& jobName,
const TString& methodTitle,
DataSetInfo& theData,
const TString& theOption,
TDirectory* theTargetDir = 0 );
const TString& theOption);
MethodDNN ( DataSetInfo& theData,
const TString& theWeightFile,
TDirectory* theTargetDir = 0 );
const TString& theWeightFile );
virtual ~MethodDNN();
@@ -131,7 +130,8 @@ namespace TMVA {
private:
TMVA::DNN::Net fNet;
std::vector<double> fWeights;
std::vector<double> fWeightBucket;
int fBucketSize;
TString fLayoutString;
std::vector<std::pair<int,TMVA::DNN::EnumFunction>> fLayout;
Oops, something went wrong.
ProTip! Use n and p to navigate between commits in a pull request.