diff --git a/extract-labels.sh b/extract-labels.sh new file mode 100755 index 0000000..34ea1f4 --- /dev/null +++ b/extract-labels.sh @@ -0,0 +1,22 @@ + + +# Number of set +k=4 + +# Vector features file +VECTOR_FILE_MASSEFFECT="data/pvectors_1rst/pvectors_teacher_${k}.txt" + +# Number of clusters +kmean=6 + +# Dirs +EXP_DIR="exp/kmeans_euclidian/teacher-pvector-1/${k}/${kmean}" +CLUSTERING="${EXP_DIR}/clustering_${kmean}.pkl" + + +# Output dirs +OUTFILE_MASSEFFECT="data/pvectors_1rst/saved_clustered/masseffect_clustered_${k}_${kmean}.txt" + +python3 bin/extract_kmeans.py "${CLUSTERING}" \ + "${VECTOR_FILE_MASSEFFECT}" \ + --outfile "$OUTFILE_MASSEFFECT"