d5f386ee8b68b01e77278f2b318de3aebbc55373,src/experiments/core.py,Experiment,model,#Experiment#,64

Before Change

        weights = datasets.load(dataset_name).class_weights
        loss = w_categorical_crossentropy(weights=weights)
        config = self.model_config
        nc = datasets.load(dataset_name).num_classes()
        w = config["dw"]
        h = config["dh"]
        autoencoder, model_name = model.build(nc=nc, w=w, h=h, loss=loss)
        if "h5file" in self.model_config:

After Change

        kwargs = self.model_config
        kwargs["nc"] = self.DatasetClass.num_classes()
        weights = self.DatasetClass.class_weights()
        kwargs["loss"] = [
            "categorical_crossentropy",
            // w_categorical_crossentropy(weights=weights)
        ]

Instances