ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
ModuleIO::Write_MLKEDF_Descriptors Class Reference

#include <write_mlkedf_descriptors.h>

Collaboration diagram for ModuleIO::Write_MLKEDF_Descriptors:

Public Member Functions

 Write_MLKEDF_Descriptors ()
 
 ~Write_MLKEDF_Descriptors ()
 
void generateTrainData_KS (const std::string &out_dir, psi::Psi< std::complex< double > > *psi, elecstate::ElecState *pelec, ModulePW::PW_Basis_K *pw_psi, ModulePW::PW_Basis *pw_rho, UnitCell &ucell, const double *veff)
 
void generateTrainData_KS (const std::string &out_dir, psi::Psi< std::complex< float > > *psi, elecstate::ElecState *pelec, ModulePW::PW_Basis_K *pw_psi, ModulePW::PW_Basis *pw_rho, UnitCell &ucell, const double *veff)
 
void generate_descriptor (const std::string &out_dir, const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< std::vector< double > > &nablaRho)
 
std::string file_name (const std::string &out_dir, std::string parameter, const int kernel_type, const double kernel_scaling)
 

Public Attributes

Cal_MLKEDF_Descriptorscal_tool = nullptr
 

Constructor & Destructor Documentation

◆ Write_MLKEDF_Descriptors()

ModuleIO::Write_MLKEDF_Descriptors::Write_MLKEDF_Descriptors ( )
inline

◆ ~Write_MLKEDF_Descriptors()

ModuleIO::Write_MLKEDF_Descriptors::~Write_MLKEDF_Descriptors ( )
inline

Member Function Documentation

◆ file_name()

std::string ModuleIO::Write_MLKEDF_Descriptors::file_name ( const std::string &  out_dir,
std::string  parameter,
const int  kernel_type,
const double  kernel_scaling 
)
Here is the caller graph for this function:

◆ generate_descriptor()

void ModuleIO::Write_MLKEDF_Descriptors::generate_descriptor ( const std::string &  out_dir,
const double *const *  prho,
ModulePW::PW_Basis pw_rho,
std::vector< std::vector< double > > &  nablaRho 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ generateTrainData_KS() [1/2]

void ModuleIO::Write_MLKEDF_Descriptors::generateTrainData_KS ( const std::string &  out_dir,
psi::Psi< std::complex< double > > *  psi,
elecstate::ElecState pelec,
ModulePW::PW_Basis_K pw_psi,
ModulePW::PW_Basis pw_rho,
UnitCell ucell,
const double *  veff 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ generateTrainData_KS() [2/2]

void ModuleIO::Write_MLKEDF_Descriptors::generateTrainData_KS ( const std::string &  out_dir,
psi::Psi< std::complex< float > > *  psi,
elecstate::ElecState pelec,
ModulePW::PW_Basis_K pw_psi,
ModulePW::PW_Basis pw_rho,
UnitCell ucell,
const double *  veff 
)
inline

Member Data Documentation

◆ cal_tool

Cal_MLKEDF_Descriptors* ModuleIO::Write_MLKEDF_Descriptors::cal_tool = nullptr

The documentation for this class was generated from the following files: