def __init__(self, network_params=None, dropout=None):
logger.info("Testeo de StackedAutoEncoder con datos de Iris")
# Datos
logger.info("Cargando datos...")
dataset = LocalLabeledDataSet()
path = os.path.abspath(os.path.join(os.path.realpath(__file__),
os.path.pardir,
os.pardir,
'examples/datasets/iris.csv'))
dataset.load_file(path)
self.train, self.valid, self.test = dataset.split_data([.5, .3, .2])
self.train = self.train.collect()
self.valid = self.valid.collect()
# Modelo
if network_params is None:
network_params = NetworkParameters(units_layers=[4, 10, 3], activation=['ReLU', 'ReLU'],
strength_l1=1e-5, strength_l2=3e-4,
dropout_ratios=[0.0, 0.0], classification=True)
self.model = StackedAutoencoder(network_params, dropout=dropout)
