for n in 3 4 5 6; do
    for ones_probability in 0.2 0.3; do
        if [ $n -eq 3 ]; then
            num_of_data=60
        elif [ $n -eq 4 ]; then
            num_of_data=120
        elif [ $n -eq 5 ]; then
            num_of_data=150
        elif [ $n -eq 6 ]; then
            num_of_data=150
        fi
        python3 -m games.tasks.kukurasu.scripts.kukurasu --num_of_data $num_of_data --n $n --m $n --ones_probability $ones_probability --max_attempts 50000
    done
done
