跳到主要内容
版本:1.1

示例

源码example目录下包含了以下示例:

执行程序可用命令如下:

PWMLFF train <input_file>
PWMLFF test <input_file>
PWMLFF explore <input_file>

其中,train表示产生数据集并训练模型,test表示测试模型,explore表示使用 DP 模型进行探索,用于使用力场模型结合 lammps 实现候选结构探索完成主动学习过程。 另外,<input_file>为 json 格式的输入文件,用于指定模型的参数、训练数据集的位置、训练模型的超参数等。

  1. Linear

  2. NN

  3. DP

提示

Linear model 中需指定 feature type 生成 descriptor,针对实际训练选择需要在 optimizer 中写不同训练组分的 weight;

NN model 与 linear model 相似,需要指定 feature type 生成 descriptor,区别在于增加了 fitting net 与具体的 optimizer(如 Adam, LKF);

DP model 的 fitting net, optimizer 与 NN model 一致,区别在于 descriptor 通过 embedding net 处理,不需要指定 feature type。

软件提供 vasp/OUTCAR 文件转换为 PWmat/MOVEMENT 文件的工具,使用方法为 outcar2movement <OUTCAR_file> <MOVEMENT_store_path>,其中<OUTCAR_file>为 vasp 的 OUTCAR 文件,<MOVEMENT_store_path>为输出的 MOVEMENT 文件的存储路径。直接运行直接读取当前路径下的 OUTCAR 文件,输出到当前路径下的 MOVEMENT 文件。