File size: 711 Bytes
fe617ac
 
 
 
 
 
 
 
71a564a
 
fe617ac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/usr/bin/expect

set host "connect.nmb2.seetacloud.com"
set port "15054"
set user "root"
set password "9Dml+WZeqp5b"
set remote_dir "/root/autodl-tmp/book-rec-with-LLMs"

# Install dependencies if needed
set cmd_pip "/root/miniconda3/bin/pip install lightgbm pandas tqdm scikit-learn"

# Run Evaluate
# We need to set PYTHONPATH because evaluation script imports src.
set cmd_eval "export PYTHONPATH=. && /root/miniconda3/bin/python scripts/evaluate.py"

spawn ssh -p $port $user@$host "cd $remote_dir; $cmd_pip; $cmd_eval"
set timeout 1200

expect {
    "password:" {
        send "$password\r"
        exp_continue
    }
    "continue connecting" {
        send "yes\r"
        exp_continue
    }
    eof
}