# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

# This is a boilerplate parameters config generated for pipeline 'eland'
# using Kedro 0.18.0.
#
# Documentation for this file format can be found in "Parameters"
# Link: https://kedro.readthedocs.io/en/0.18.0/kedro_project_setup/configuration.html#parameters
eland_data_load_options:
  dataset: 'reddit'
  baseline: 'store_true'
  batch_size: 300
  #graph_num: 3

eland_model_options:
  dim_feats: 300
  cuda: 0
  hidden_size: 128
  n_layers: 2
  epochs: 50
  batch_size: 300
  seed: -1
  lr: 0.0001
  log: True
  weight_decay: 0.000001
  dropout: 0.4
  tensorboard: False
  log: True
  name: 'debug' 
  gnnlayer_type: 'gcn'
  rnn_type: 'lstm'
  pretrain_bm: 25 
  pretrain_nc: 200
  alpha: 0.05
  bmloss_type: 'mse'
  device: 'cpu'
  base_pred: 400
  save_directory: "data/07_model_output/user_behavior"