search

sidhomj / DeepTCR

Deep Learning Methods for Parsing T-Cell Receptor Sequencing (TCRSeq) Data
https://sidhomj.github.io/DeepTCR/
MIT License
113 stars 40 forks source link

parallele for KNN #7

Closed hejing3283 closed 5 years ago

hejing3283 commented 5 years ago

Could we add n_jobs to the KNN_Sequence_Classifier by adding n_jobs to KNeighborsClassifier from sklearn module? Currently the KNN_Sequence_Classifier is very slow.

Thanks!

sidhomj commented 5 years ago

sure thing but if I remember correctly, when I've tried to use the KNeighborsClassifier in parallel, it weirdly ends up being slower but I can add it in in there.

sidhomj commented 5 years ago

Just pushed the updated to parallelize into the method to the master branch. Let me know if you if you have any problems with it!

hejing3283 commented 5 years ago

Thanks! I will let you know after I finish running.