Skip to content

Commit 7fe1d81

Browse files
seq_emb.sh
1 parent 5a7a879 commit 7fe1d81

1 file changed

Lines changed: 6 additions & 18 deletions

File tree

scripts/aido/seq_emb.sh

Lines changed: 6 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,17 @@
11
#!/bin/bash
22

3-
# script used to score Nucleotide Transformer model
4-
export reference_sheet="/home/ma_run_ze/lzm/rnagym/fitness/reference_sheet_final.csv"
5-
export dms_data_dir="/home/ma_run_ze/lzm/rnagym/fitness/fitness_processed_assays"
3+
4+
export reference_sheet=""
5+
export dms_data_dir=""
66
mkdir -p "$output_scores_dir"
7-
# https://huggingface.co/InstaDeepAI/nucleotide-transformer-2.5b-multi-species
8-
DMS_index=$((44)) # Change this to the desired DMS index
9-
export HF_ENDPOINT=https://hf-mirror.com
7+
DMS_index=$((44))
108
source /data4/marunze/aido/bin/activate
119

12-
export CUDA_VISIBLE_DEVICES=1 # Set to the GPU you want to use
13-
# Remove existing score files and create a new summary file
14-
# Get the current index from the array (0-31)
15-
16-
1710

18-
19-
export output_dir="/data_share/marunze/lzm/rnagym/fitness/embeddings/aido/"
20-
export dms_data_dir="/home/ma_run_ze/lzm/rnagym/fitness/fitness_processed_assays"
11+
export output_dir=""
12+
export dms_data_dir=""
2113
mkdir -p "$output_dir"
22-
# Get the current index from the array (0-31)
23-
24-
# Run the scoring script with the array task ID
2514
python seq_emb.py \
2615
--output_dir_path "$output_dir" \
2716
--ref_sheet "$reference_sheet" \
2817
--dms_dir_path "$dms_data_dir" \
29-

0 commit comments

Comments
 (0)