#!/bin/bash
conda activate torch

dataset=enron
for authors in 10 50 ;  do
  for alpha in 1 0.999 0.99 0.9 0.5 ; do
    python main.py --dataset $dataset -A $authors -s 0 -o ../results_$dataset.csv --alpha $alpha
  done
done

dataset=imdb62
for alpha in 1 0.999 0.99 0.9 0.5 ; do
  python main.py --dataset $dataset -A -1 -s 0 -o ../results_$dataset.csv --alpha $alpha
done