chemprop.utils.v1_to_v2

Attributes

logger

Functions

max_encoder_index(d)
convert_state_dict_v1_to_v2(model_v1_dict)	Converts v1 model dictionary to a v2 state dictionary
convert_hyper_parameters_v1_to_v2(model_v1_dict)	Converts v1 model dictionary to v2 hyper_parameters dictionary
convert_model_dict_v1_to_v2(model_v1_dict)	Converts a v1 model dictionary from a loaded .pt file to a v2 model dictionary
convert_model_file_v1_to_v2(model_v1_file, model_v2_file)	Converts a v1 model .pt file to a v2 model .pt file

Module Contents

chemprop.utils.v1_to_v2.logger

chemprop.utils.v1_to_v2.max_encoder_index(d)

Parameters:

d (dict)

Return type:

int

chemprop.utils.v1_to_v2.convert_state_dict_v1_to_v2(model_v1_dict)

Converts v1 model dictionary to a v2 state dictionary

Parameters:

model_v1_dict (dict)

Return type:

dict

chemprop.utils.v1_to_v2.convert_hyper_parameters_v1_to_v2(model_v1_dict)

Converts v1 model dictionary to v2 hyper_parameters dictionary

Parameters:

model_v1_dict (dict)

Return type:

dict

chemprop.utils.v1_to_v2.convert_model_dict_v1_to_v2(model_v1_dict)

Converts a v1 model dictionary from a loaded .pt file to a v2 model dictionary

Parameters:

model_v1_dict (dict)

Return type:

dict

chemprop.utils.v1_to_v2.convert_model_file_v1_to_v2(model_v1_file, model_v2_file)

Converts a v1 model .pt file to a v2 model .pt file

Parameters:

• model_v1_file (os.PathLike)

• model_v2_file (os.PathLike)

Return type:

None