How to use scikit learn train test split












0












$begingroup$


Is there a command within scikit-learn that allows for splitting based on participant? I have 50 participants x 90 trials each, but we want to test on unseen data as a way of assessing overfit.






machine-learning







share|improve this question









New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.







$endgroup$












  • $begingroup$
    What do you mean by "splitting based on participant?"
    $endgroup$
    – Ethan
    11 hours ago










  • $begingroup$
    @Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
    $endgroup$
    – Andrew
    8 hours ago
















0












$begingroup$


Is there a command within scikit-learn that allows for splitting based on participant? I have 50 participants x 90 trials each, but we want to test on unseen data as a way of assessing overfit.






machine-learning







share|improve this question









New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.







$endgroup$












  • $begingroup$
    What do you mean by "splitting based on participant?"
    $endgroup$
    – Ethan
    11 hours ago










  • $begingroup$
    @Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
    $endgroup$
    – Andrew
    8 hours ago














0












0








0





$begingroup$


Is there a command within scikit-learn that allows for splitting based on participant? I have 50 participants x 90 trials each, but we want to test on unseen data as a way of assessing overfit.






machine-learning







share|improve this question









New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.







$endgroup$




Is there a command within scikit-learn that allows for splitting based on participant? I have 50 participants x 90 trials each, but we want to test on unseen data as a way of assessing overfit.






machine-learning




machine-learning






share|improve this question









New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.











share|improve this question









New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.









share|improve this question




share|improve this question








edited 8 mins ago









Ethan

566222




566222






New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.









asked 11 hours ago









AndrewAndrew

1




1




New contributor




Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.





New contributor





Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.






Andrew is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.












  • $begingroup$
    What do you mean by "splitting based on participant?"
    $endgroup$
    – Ethan
    11 hours ago










  • $begingroup$
    @Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
    $endgroup$
    – Andrew
    8 hours ago


















  • $begingroup$
    What do you mean by "splitting based on participant?"
    $endgroup$
    – Ethan
    11 hours ago










  • $begingroup$
    @Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
    $endgroup$
    – Andrew
    8 hours ago
















$begingroup$
What do you mean by "splitting based on participant?"
$endgroup$
– Ethan
11 hours ago




$begingroup$
What do you mean by "splitting based on participant?"
$endgroup$
– Ethan
11 hours ago












$begingroup$
@Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
$endgroup$
– Andrew
8 hours ago




$begingroup$
@Ethan As in, I would like to split based on participant IDs and then classify exercises. For example, if I wanted to do a 50/50 train/test split, then I would have 25 participants data in the training, then test on the other 25 participants data.
$endgroup$
– Andrew
8 hours ago










1 Answer
1






active

oldest

votes


















1












$begingroup$

To split data into Training and Test sets in scikit-learn you can use the train_test_split function.



To implement this:



from sklearn.model_selection import train_test_split



sklearn.model_selection.train_test_split()


Then pass through the arguments that you would like to set as parameters. Some parameters of note you may like to tune are the sizes of the training and test sets as well as setting if the splits are shuffled or not.



You can read about this further on its scikit-learn documentation page here.






share|improve this answer









$endgroup$













  • $begingroup$
    Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
    $endgroup$
    – Andrew
    6 hours ago










  • $begingroup$
    train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
    $endgroup$
    – Ethan
    4 hours ago












  • $begingroup$
    No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
    $endgroup$
    – Andrew
    38 mins ago










  • $begingroup$
    I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
    $endgroup$
    – Andrew
    5 mins ago











Your Answer





StackExchange.ifUsing("editor", function () {
return StackExchange.using("mathjaxEditing", function () {
StackExchange.MarkdownEditor.creationCallbacks.add(function (editor, postfix) {
StackExchange.mathjaxEditing.prepareWmdForMathJax(editor, postfix, [["$", "$"], ["\\(","\\)"]]);
});
});
}, "mathjax-editing");

StackExchange.ready(function() {
var channelOptions = {
tags: "".split(" "),
id: "557"
};
initTagRenderer("".split(" "), "".split(" "), channelOptions);

StackExchange.using("externalEditor", function() {
// Have to fire editor after snippets, if snippets enabled
if (StackExchange.settings.snippets.snippetsEnabled) {
StackExchange.using("snippets", function() {
createEditor();
});
}
else {
createEditor();
}
});

function createEditor() {
StackExchange.prepareEditor({
heartbeatType: 'answer',
autoActivateHeartbeat: false,
convertImagesToLinks: false,
noModals: true,
showLowRepImageUploadWarning: true,
reputationToPostImages: null,
bindNavPrevention: true,
postfix: "",
imageUploader: {
brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
allowUrls: true
},
onDemand: true,
discardSelector: ".discard-answer"
,immediatelyShowMarkdownHelp:true
});


}
});






Andrew is a new contributor. Be nice, and check out our Code of Conduct.










draft saved

draft discarded


















StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fdatascience.stackexchange.com%2fquestions%2f47544%2fhow-to-use-scikit-learn-train-test-split%23new-answer', 'question_page');
}
);

Post as a guest















Required, but never shown

























1 Answer
1






active

oldest

votes








1 Answer
1






active

oldest

votes









active

oldest

votes






active

oldest

votes









1












$begingroup$

To split data into Training and Test sets in scikit-learn you can use the train_test_split function.



To implement this:



from sklearn.model_selection import train_test_split



sklearn.model_selection.train_test_split()


Then pass through the arguments that you would like to set as parameters. Some parameters of note you may like to tune are the sizes of the training and test sets as well as setting if the splits are shuffled or not.



You can read about this further on its scikit-learn documentation page here.






share|improve this answer









$endgroup$













  • $begingroup$
    Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
    $endgroup$
    – Andrew
    6 hours ago










  • $begingroup$
    train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
    $endgroup$
    – Ethan
    4 hours ago












  • $begingroup$
    No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
    $endgroup$
    – Andrew
    38 mins ago










  • $begingroup$
    I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
    $endgroup$
    – Andrew
    5 mins ago
















1












$begingroup$

To split data into Training and Test sets in scikit-learn you can use the train_test_split function.



To implement this:



from sklearn.model_selection import train_test_split



sklearn.model_selection.train_test_split()


Then pass through the arguments that you would like to set as parameters. Some parameters of note you may like to tune are the sizes of the training and test sets as well as setting if the splits are shuffled or not.



You can read about this further on its scikit-learn documentation page here.






share|improve this answer









$endgroup$













  • $begingroup$
    Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
    $endgroup$
    – Andrew
    6 hours ago










  • $begingroup$
    train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
    $endgroup$
    – Ethan
    4 hours ago












  • $begingroup$
    No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
    $endgroup$
    – Andrew
    38 mins ago










  • $begingroup$
    I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
    $endgroup$
    – Andrew
    5 mins ago














1












1








1





$begingroup$

To split data into Training and Test sets in scikit-learn you can use the train_test_split function.



To implement this:



from sklearn.model_selection import train_test_split



sklearn.model_selection.train_test_split()


Then pass through the arguments that you would like to set as parameters. Some parameters of note you may like to tune are the sizes of the training and test sets as well as setting if the splits are shuffled or not.



You can read about this further on its scikit-learn documentation page here.






share|improve this answer









$endgroup$



To split data into Training and Test sets in scikit-learn you can use the train_test_split function.



To implement this:



from sklearn.model_selection import train_test_split



sklearn.model_selection.train_test_split()


Then pass through the arguments that you would like to set as parameters. Some parameters of note you may like to tune are the sizes of the training and test sets as well as setting if the splits are shuffled or not.



You can read about this further on its scikit-learn documentation page here.







share|improve this answer












share|improve this answer



share|improve this answer










answered 11 hours ago









EthanEthan

566222




566222












  • $begingroup$
    Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
    $endgroup$
    – Andrew
    6 hours ago










  • $begingroup$
    train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
    $endgroup$
    – Ethan
    4 hours ago












  • $begingroup$
    No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
    $endgroup$
    – Andrew
    38 mins ago










  • $begingroup$
    I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
    $endgroup$
    – Andrew
    5 mins ago


















  • $begingroup$
    Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
    $endgroup$
    – Andrew
    6 hours ago










  • $begingroup$
    train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
    $endgroup$
    – Ethan
    4 hours ago












  • $begingroup$
    No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
    $endgroup$
    – Andrew
    38 mins ago










  • $begingroup$
    I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
    $endgroup$
    – Andrew
    5 mins ago
















$begingroup$
Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
$endgroup$
– Andrew
6 hours ago




$begingroup$
Yes, I am using that function. I don't think stratify is the correct parameter since all that is doing it looking to make sure proportions stay the same. I just want to split participant data as being used for training only or testing only.
$endgroup$
– Andrew
6 hours ago












$begingroup$
train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
$endgroup$
– Ethan
4 hours ago






$begingroup$
train_test_split will by default, do a 75% training 25% testing split. Is this what you want to change? I am not sure I understand your question.
$endgroup$
– Ethan
4 hours ago














$begingroup$
No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
$endgroup$
– Andrew
38 mins ago




$begingroup$
No, I want to split based on participants. Currently it randomly splits so all 50 participants are very likely going to be included in the training and testing portions.
$endgroup$
– Andrew
38 mins ago












$begingroup$
I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
$endgroup$
– Andrew
5 mins ago




$begingroup$
I want to do a 50% train split with the additional parameter that participants are mutually excluded to either the train or test split. So there are 25 participants worth of data on the train side and 25 participants worth of data on the test side. Basically, is there an easy way to randomize 50 participants into two groups? It seems like the answer is no and requires a custom script.
$endgroup$
– Andrew
5 mins ago










Andrew is a new contributor. Be nice, and check out our Code of Conduct.










draft saved

draft discarded


















Andrew is a new contributor. Be nice, and check out our Code of Conduct.













Andrew is a new contributor. Be nice, and check out our Code of Conduct.












Andrew is a new contributor. Be nice, and check out our Code of Conduct.
















Thanks for contributing an answer to Data Science Stack Exchange!


  • Please be sure to answer the question. Provide details and share your research!

But avoid …



  • Asking for help, clarification, or responding to other answers.

  • Making statements based on opinion; back them up with references or personal experience.


Use MathJax to format equations. MathJax reference.


To learn more, see our tips on writing great answers.




draft saved


draft discarded














StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fdatascience.stackexchange.com%2fquestions%2f47544%2fhow-to-use-scikit-learn-train-test-split%23new-answer', 'question_page');
}
);

Post as a guest















Required, but never shown





















































Required, but never shown














Required, but never shown












Required, but never shown







Required, but never shown

































Required, but never shown














Required, but never shown












Required, but never shown







Required, but never shown







-

Popular posts from this blog

Ponta tanko

Image
Usona ponta tanko M60A1 Ponta tanko estas, simile kiel kirasita savveturilo, inter specialaj kirasitaj transportiloj. Kiel pontaj tankoj eblas konsideri ĉiujn memmovigaj rimedoj ebligantaj superi obstaklojn ĝis larĝeco 24 metroj rekte en spaco de bataloperacoj. Ties ĉefa tasko estas ebligi al la propraj taĉmentoj daŭrigi en rapida antaŭenirado sen senbezona reteniĝo. El tio pli-malpli rezultas taktikaj-teknikaj postuloj, kiujn ili devas plenumi. Precipe temas pri kapablo operacii komune kun unue-sinsekvaj taĉmentoj kaj transpontigi obstaklojn dum kelke da minutoj. This page is only for reference, If you need detailed information, please check here
Read more

Tantalo (mitologio)

Image
Ĉi tiu artikolo temas pri mitologio. Por aliaj signifoj vidu la paĝon Tantalo (apartigilo) . Tantalo , laŭ Gioacchino Assereto. Laŭ la helena mitologio, Tantalo (el la greka Τάνταλος , ŝancelanta aŭ tre malfeliĉa ) estis filo de Zeŭso kaj la oceanidino Pluto, laŭ la plej divastigata versio. Li estis reĝo de Frigio [1] aŭ de la monto Sipilo, en Lidio (Malgrandazio) [2] kaj post sia morto li suferis eternan torturadon ĉe la Tartaro, la plej profunda parto de Hadeso, rezervita por la plej perversaj homoj. Tamen, kelkaj rakontoj asertas ke li estis reĝo de Argo aŭ de Korinto [3] , aŭ ke li migris norden el la monto Sipilo por akiri la tronon de Paflagonio, el kie Ilo forpelis lin pro tio, ke Tantalo forkaptis kaj seksperfortis lian fraton Ganimedon [4] , kaj tiel ofendis la diojn. Verŝajne li ankaŭ regis urbon en okcidenta Anatolio, kiun oni nomis Tantalis [5] pro li, aŭ Sipilo , pro la monto sur kies flanko ankoraŭ staras ĝiaj ruinaĵoj [6] . Laŭ la geografiisto Strabo,...
Read more

Erzsébet Schaár

Image
Erzsébet Schaár Memportreto de Erzsébet Schaár en Pécs Personaj informoj Naskiĝo 27-an de julio 1908 ( 1908-07-27 ) en Budafok Morto 29-an de aŭgusto 1975 ( 1975-08-29 ) (67-jara) en Budapeŝto Ŝtataneco Hungario Profesio skulptisto Familianoj Edz(in)o Tibor Vilt v   •   d   •   r Erzsébet Schaár [erĵEbet ŝAr], laŭ hungarlingve kutima sinsekvo Schaár Erzsébet estis hungara skulptisto. Ŝia edzo estis Tibor Vilt. Déryné Széppataki Róza en Miskolc fare de Erzsébet Schaár Erzsébet Schaár [1] naskiĝis la 27-an de julio 1908 en Budafok, ŝi mortis la 29-an de aŭgusto 1975 en Budapeŝto. Enhavo 1 Biografio 2 Elektitaj solaj ekspozicioj 3 Elektitaj kolektivaj ekspozicioj 4 Elektitaj skulptaĵoj 5 Elektitaj disĉiploj 6 Fontoj 7 Referencoj Biografio | Erzsébet Schaár studis en Industriarta Lernejo, poste en Reĝa Belarta Altlernejo (1924-1926). Ŝia ĉefinstruisto estis Zsig...
Read more