Update with adaptations on ptt paper #53

Merged

nbereux merged 86 commits intodevelopfrom

papier_ptt_train

Mar 4, 2026

Commits on Oct 17, 2025

save code
Nicolas Béreux
committed
remove warmup
Nicolas Béreux
committed

Commits on Nov 4, 2025

compile normalize grad + figsize factor on plot
Nicolas Béreux
committed

Commits on Nov 17, 2025

make remove duplicates optional
Nicolas Béreux
committed

Commits on Dec 1, 2025

fix prev_size dataset in split_data script
Nicolas Béreux
committed

Commits on Dec 15, 2025

format code
Nicolas Béreux
committed

Commits on Dec 16, 2025

add map model as option
Nicolas Béreux
committed

Commits on Dec 19, 2025

Merge branch 'develop' of github.com:DsysDML/rbms into papier_ptt_train
Nicolas Béreux
committed
Merge branch 'develop' of github.com:DsysDML/rbms into papier_ptt_train
Nicolas Béreux
committed
Merge branch 'develop' of github.com:DsysDML/rbms into papier_ptt_train
Nicolas Béreux
committed
fix merge
Nicolas Béreux
committed
remove 3.14 as torch compile is not supported yet
Nicolas Béreux
committed
fix merge
Nicolas Béreux
committed
add missing keys to args dict in tests
Nicolas Béreux
committed

Commits on Jan 9, 2026

batch function dataset
Nicolas Béreux
committed
Merge branch 'develop' of github.com:DsysDML/rbms into papier_ptt_train
Nicolas Béreux
committed
use batch method
Nicolas Béreux
committed

Commits on Jan 13, 2026

add visible_type to EBM class
Nicolas Béreux
committed
change variable_type after conversion
Nicolas Béreux
committed
change variable_type from binary to bernoulli
Nicolas Béreux
committed
add visible_type
Nicolas Béreux
committed
add categorical_to_bernoulli implementation
Nicolas Béreux
committed
fix variable_type
Nicolas Béreux
committed
match dataset variable type with model visible type
Nicolas Béreux
committed

Commits on Jan 14, 2026

sample bernoulli when variable_type is bernoulli
Nicolas Béreux
committed
add log_scale option to PCA plot
Nicolas Béreux
committed

Commits on Jan 27, 2026

removed unused variable in non centered gradient
Nicolas Béreux
committed
add conversion print + astype to dataset class
Nicolas Béreux
committed
add __eq__ to class for easier comparison
Nicolas Béreux
committed
add IIRBM and BGRBM to map_model
Nicolas Béreux
committed
add model_type and normalize_grad option to parser
Nicolas Béreux
committed
add dataset weights arg
Nicolas Béreux
committed
fix binary to bernoulli and add ising to model match
Nicolas Béreux
committed
make normalize_grad optional
Nicolas Béreux
committed
save result from get_eigenvalues_history in file to avoid repeating computations
Nicolas Béreux
committed
change version number
Nicolas Béreux
committed
simplify imports
Nicolas Béreux
committed

Commits on Jan 29, 2026

fix: add __init__ to bernoulli_gaussian
Nicolas Béreux
committed

Commits on Feb 4, 2026

clip grad
Nicolas Béreux
committed
rework the main loop and add rbms restore script allowing to change many hyperparameters
Nicolas Béreux
committed
new parser, keep the old fucntions for compatibility
Nicolas Béreux
committed
save learning rate during training and remove the hyperparameters loading from load_model
Nicolas Béreux
committed
util to handle optimizer declaration
Nicolas Béreux
committed
remove test for removed function
Nicolas Béreux
committed
remove weights from init_parameters
Nicolas Béreux
committed
add learning_rate
Nicolas Béreux
committed

Commits on Feb 11, 2026

margaret update
Nicolas Béreux
committed

Commits on Feb 18, 2026

remove prefactor variance initialization w
Nicolas Béreux
committed
formatting
Nicolas Béreux
committed
fix gzip
Nicolas Béreux
committed
add Sampler class for more training modularity
Nicolas Béreux
committed
add cossim optimizer
Nicolas Béreux
committed
put gradient modifications in one pipeline
Nicolas Béreux
committed
cd, pcd and rdm sampler
Nicolas Béreux
committed
use new training function and script
Nicolas Béreux
committed
fix some compatibility for restore
Nicolas Béreux
committed

Commits on Feb 19, 2026

Merge branch 'develop' of github.com:DsysDML/rbms into papier_ptt_train
Nicolas Béreux
committed
put all saving methods with np array
Nicolas Béreux
committed

Commits on Feb 20, 2026

change sampler call during training to get_grad_conf and add post_grad_update to EBM
Nicolas Béreux
committed
add check keys dict + save sampler name
Nicolas Béreux
committed

Commits on Feb 25, 2026

type hinting
Nicolas Béreux
committed
metrics save and display for sampler
Nicolas Béreux
committed
remove unused import
Nicolas Béreux
committed
type hint
Nicolas Béreux
committed
add some safeguard against None
Nicolas Béreux
committed
correct type hint for map
Nicolas Béreux
committed
regularization is now handled by pre_grad_update
Nicolas Béreux
committed

Commits on Feb 26, 2026

remove compilation
Nicolas Béreux
committed
remove unused code + type hint
Nicolas Béreux
committed

Commits on Mar 2, 2026

fix format of default names when splitting data
Nicolas Béreux
committed

Commits on Mar 4, 2026

save parallel chains as bool/int
Nicolas Béreux
committed
fix read training_type
Nicolas Béreux
committed
update h5py version
Nicolas Béreux
committed