mechanoChemML.src.hparameters_cnn_grid

Module Contents

Classes

HyperParametersCNN This class is created to perform hyper-parameters search for CNNs.

Functions

getlist_str(option, sep=’,’, chars=None) Return a list from a ConfigParser option. By default,
getlist_int(option, sep=’,’, chars=None) Return a list from a ConfigParser option. By default,
mechanoChemML.src.hparameters_cnn_grid.getlist_str(option, sep=', ', chars=None)[source]

Return a list from a ConfigParser option. By default, split on a comma and strip whitespaces.

mechanoChemML.src.hparameters_cnn_grid.getlist_int(option, sep=', ', chars=None)[source]

Return a list from a ConfigParser option. By default, split on a comma and strip whitespaces.

class mechanoChemML.src.hparameters_cnn_grid.HyperParametersCNN(config, input_shape=1, output_shape=1, filter_size=3, pool_size=1, padding='same', dim=2, uniform_sample_number=10, neighbor_sample_number=5, iteration_time=3, sample_ratio=0.3, best_model_number=10, max_total_parameter=100000.0, repeat_train=3, debug=False)[source]

This class is created to perform hyper-parameters search for CNNs.

initialize_info(self)[source]
update_model_info(self, index0, history)[source]

update the summary for a particular CNN

compute_total_parameters_per_CNN(self, filters)[source]

if pool_size == 1, no maxpooling layer, if pool_size > 1, pooling layer will be inserted if outputlayer = dense, a flatten layer will be inserted

generate_all_CNNs(self)[source]
do_the_sampling(self)[source]
update_index_list(self)[source]
get_next_model(self)[source]
plot_best_models(self, model_number_to_plot=5)[source]
plot_all_models(self)[source]
mechanoChemML.src.hparameters_cnn_grid.config =[source]