TestBackpropagationCpu.cxx

Go to the documentation of this file.

// @(#)root/tmva $Id$
// Author: Simon Pfreundschuh

/*************************************************************************
 * Copyright (C) 2016, Simon Pfreundschuh                                *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

////////////////////////////////////////////////////////////////////
// Concrete instantiation of the generic backpropagation test for //
// multi-threaded CPU architectures.                              //
////////////////////////////////////////////////////////////////////

#include "TMatrix.h"
#include "TMVA/DNN/Architectures/Cpu.h"
#include "TestBackpropagation.h"

using namespace TMVA::DNN;

int main()
{
   using Scalar_t = Double_t;
   std::cout << "Testing Backpropagation:" << std::endl;

   double error;

   error = testBackpropagationWeightsLinear<TCpu<Scalar_t>>(1.0);
   if (error > 1e-3)
       return 1;

   error = testBackpropagationL1Regularization<TCpu<Scalar_t>>(1e-2);
   if (error > 1e-3)
       return 1;

   error = testBackpropagationL2Regularization<TCpu<Scalar_t>>(1.0);
   if (error > 1e-3)
       return 1;

   error = testBackpropagationBiasesLinear<TCpu<Scalar_t>>(1.0);
   if (error > 1e-3)
       return 1;

   return 0;
}