{"id":10647,"date":"2024-03-14T12:29:13","date_gmt":"2024-03-14T12:29:13","guid":{"rendered":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/"},"modified":"2025-08-04T02:08:02","modified_gmt":"2025-08-04T02:08:02","slug":"how-to-use-the-train_test_split-function-in-python","status":"publish","type":"post","link":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/","title":{"rendered":"Python train_test_split Guide"},"content":{"rendered":"<p>The train_test_split function is used to divide a dataset into training and testing sets. In Python, the train_test_split function can be accessed through the model_selection module in the sklearn library. Here is a simple example:<\/p>\n<pre class=\"post-pre\"><code><span class=\"hljs-keyword\">from<\/span> sklearn.model_selection <span class=\"hljs-keyword\">import<\/span> train_test_split\r\n<span class=\"hljs-keyword\">import<\/span> numpy <span class=\"hljs-keyword\">as<\/span> np\r\n\r\n<span class=\"hljs-comment\"># \u751f\u6210\u4e00\u4e9b\u793a\u4f8b\u6570\u636e<\/span>\r\nX = np.array([[<span class=\"hljs-number\">1<\/span>, <span class=\"hljs-number\">2<\/span>], [<span class=\"hljs-number\">3<\/span>, <span class=\"hljs-number\">4<\/span>], [<span class=\"hljs-number\">5<\/span>, <span class=\"hljs-number\">6<\/span>], [<span class=\"hljs-number\">7<\/span>, <span class=\"hljs-number\">8<\/span>]])\r\ny = np.array([<span class=\"hljs-number\">0<\/span>, <span class=\"hljs-number\">1<\/span>, <span class=\"hljs-number\">0<\/span>, <span class=\"hljs-number\">1<\/span>])\r\n\r\n<span class=\"hljs-comment\"># \u5c06\u6570\u636e\u96c6\u5212\u5206\u4e3a\u8bad\u7ec3\u96c6\u548c\u6d4b\u8bd5\u96c6\uff0ctest_size\u6307\u5b9a\u6d4b\u8bd5\u96c6\u6bd4\u4f8b\uff0crandom_state\u8bbe\u7f6e\u968f\u673a\u79cd\u5b50<\/span>\r\nX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=<span class=\"hljs-number\">0.2<\/span>, random_state=<span class=\"hljs-number\">42<\/span>)\r\n\r\n<span class=\"hljs-comment\"># \u6253\u5370\u5212\u5206\u540e\u7684\u8bad\u7ec3\u96c6\u548c\u6d4b\u8bd5\u96c6<\/span>\r\n<span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">\"\u8bad\u7ec3\u96c6X:\"<\/span>, X_train)\r\n<span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">\"\u6d4b\u8bd5\u96c6X:\"<\/span>, X_test)\r\n<span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">\"\u8bad\u7ec3\u96c6y:\"<\/span>, y_train)\r\n<span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">\"\u6d4b\u8bd5\u96c6y:\"<\/span>, y_test)\r\n<\/code><\/pre>\n<p>In this example, we created a dataset X containing 4 samples and their corresponding labels y. Then, we used the train_test_split function to divide the dataset into a training set and a test set, with the test_size parameter specifying the proportion of the test set (in this case, 0.2 or 20%), and the random_state parameter setting a random seed to ensure consistent results in the split. Finally, we printed out the training and test sets after the division.<\/p>\n","protected":false},"excerpt":{"rendered":"<p>The train_test_split function is used to divide a dataset into training and testing sets. In Python, the train_test_split function can be accessed through the model_selection module in the sklearn library. Here is a simple example: from sklearn.model_selection import train_test_split import numpy as np # \u751f\u6210\u4e00\u4e9b\u793a\u4f8b\u6570\u636e X = np.array([[1, 2], [3, 4], [5, 6], [7, 8]]) [&hellip;]<\/p>\n","protected":false},"author":13,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"_import_markdown_pro_load_document_selector":0,"_import_markdown_pro_submit_text_textarea":"","footnotes":""},"categories":[1],"tags":[8535,75,72,13394,13384],"class_list":["post-10647","post","type-post","status-publish","format-standard","hentry","category-uncategorized","tag-data-splitting","tag-machine-learning","tag-python","tag-sklearn","tag-train_test_split"],"yoast_head":"<!-- This site is optimized with the Yoast SEO Premium plugin v21.5 (Yoast SEO v21.5) - https:\/\/yoast.com\/wordpress\/plugins\/seo\/ -->\n<title>Python train_test_split Guide - Blog - Silicon Cloud<\/title>\n<meta name=\"description\" content=\"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.\" \/>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\" \/>\n<meta property=\"og:locale\" content=\"en_US\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"Python train_test_split Guide\" \/>\n<meta property=\"og:description\" content=\"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.\" \/>\n<meta property=\"og:url\" content=\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\" \/>\n<meta property=\"og:site_name\" content=\"Blog - Silicon Cloud\" \/>\n<meta property=\"article:publisher\" content=\"https:\/\/www.facebook.com\/SiliCloudGlobal\/\" \/>\n<meta property=\"article:published_time\" content=\"2024-03-14T12:29:13+00:00\" \/>\n<meta property=\"article:modified_time\" content=\"2025-08-04T02:08:02+00:00\" \/>\n<meta name=\"author\" content=\"Isabella Edwards\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:creator\" content=\"@SiliCloudGlobal\" \/>\n<meta name=\"twitter:site\" content=\"@SiliCloudGlobal\" \/>\n<meta name=\"twitter:label1\" content=\"Written by\" \/>\n\t<meta name=\"twitter:data1\" content=\"Isabella Edwards\" \/>\n\t<meta name=\"twitter:label2\" content=\"Est. reading time\" \/>\n\t<meta name=\"twitter:data2\" content=\"1 minute\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"Article\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#article\",\"isPartOf\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\"},\"author\":{\"name\":\"Isabella Edwards\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/5579144e23c225c8188167f3e3f888dd\"},\"headline\":\"Python train_test_split Guide\",\"datePublished\":\"2024-03-14T12:29:13+00:00\",\"dateModified\":\"2025-08-04T02:08:02+00:00\",\"mainEntityOfPage\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\"},\"wordCount\":124,\"publisher\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\"},\"keywords\":[\"data splitting\",\"machine learning\",\"Python\",\"sklearn\",\"train_test_split\"],\"inLanguage\":\"en-US\"},{\"@type\":\"WebPage\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\",\"url\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\",\"name\":\"Python train_test_split Guide - Blog - Silicon Cloud\",\"isPartOf\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#website\"},\"datePublished\":\"2024-03-14T12:29:13+00:00\",\"dateModified\":\"2025-08-04T02:08:02+00:00\",\"description\":\"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.\",\"breadcrumb\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#breadcrumb\"},\"inLanguage\":\"en-US\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/\"]}]},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"Home\",\"item\":\"https:\/\/www.silicloud.com\/blog\/\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"Python train_test_split Guide\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#website\",\"url\":\"https:\/\/www.silicloud.com\/blog\/\",\"name\":\"Silicon Cloud Blog\",\"description\":\"\",\"publisher\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\"},\"inLanguage\":\"en-US\"},{\"@type\":\"Organization\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\",\"name\":\"Silicon Cloud Blog\",\"url\":\"https:\/\/www.silicloud.com\/blog\/\",\"logo\":{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/\",\"url\":\"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png\",\"contentUrl\":\"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png\",\"width\":1024,\"height\":1024,\"caption\":\"Silicon Cloud Blog\"},\"image\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/\"},\"sameAs\":[\"https:\/\/www.facebook.com\/SiliCloudGlobal\/\",\"https:\/\/twitter.com\/SiliCloudGlobal\"]},{\"@type\":\"Person\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/5579144e23c225c8188167f3e3f888dd\",\"name\":\"Isabella Edwards\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/image\/\",\"url\":\"https:\/\/secure.gravatar.com\/avatar\/d4d4dec47f553ac7961d9fa4cc9bdcdcf5b7ce5106594330b6d25c5694fdbaec?s=96&d=mm&r=g\",\"contentUrl\":\"https:\/\/secure.gravatar.com\/avatar\/d4d4dec47f553ac7961d9fa4cc9bdcdcf5b7ce5106594330b6d25c5694fdbaec?s=96&d=mm&r=g\",\"caption\":\"Isabella Edwards\"},\"url\":\"https:\/\/www.silicloud.com\/blog\/author\/isabellaedwards\/\"}]}<\/script>\n<!-- \/ Yoast SEO Premium plugin. -->","yoast_head_json":{"title":"Python train_test_split Guide - Blog - Silicon Cloud","description":"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/","og_locale":"en_US","og_type":"article","og_title":"Python train_test_split Guide","og_description":"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.","og_url":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/","og_site_name":"Blog - Silicon Cloud","article_publisher":"https:\/\/www.facebook.com\/SiliCloudGlobal\/","article_published_time":"2024-03-14T12:29:13+00:00","article_modified_time":"2025-08-04T02:08:02+00:00","author":"Isabella Edwards","twitter_card":"summary_large_image","twitter_creator":"@SiliCloudGlobal","twitter_site":"@SiliCloudGlobal","twitter_misc":{"Written by":"Isabella Edwards","Est. reading time":"1 minute"},"schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"Article","@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#article","isPartOf":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/"},"author":{"name":"Isabella Edwards","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/5579144e23c225c8188167f3e3f888dd"},"headline":"Python train_test_split Guide","datePublished":"2024-03-14T12:29:13+00:00","dateModified":"2025-08-04T02:08:02+00:00","mainEntityOfPage":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/"},"wordCount":124,"publisher":{"@id":"https:\/\/www.silicloud.com\/blog\/#organization"},"keywords":["data splitting","machine learning","Python","sklearn","train_test_split"],"inLanguage":"en-US"},{"@type":"WebPage","@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/","url":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/","name":"Python train_test_split Guide - Blog - Silicon Cloud","isPartOf":{"@id":"https:\/\/www.silicloud.com\/blog\/#website"},"datePublished":"2024-03-14T12:29:13+00:00","dateModified":"2025-08-04T02:08:02+00:00","description":"Learn how to use train_test_split in Python to split datasets for machine learning. Includes examples and parameter explanations.","breadcrumb":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#breadcrumb"},"inLanguage":"en-US","potentialAction":[{"@type":"ReadAction","target":["https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/"]}]},{"@type":"BreadcrumbList","@id":"https:\/\/www.silicloud.com\/blog\/how-to-use-the-train_test_split-function-in-python\/#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"Home","item":"https:\/\/www.silicloud.com\/blog\/"},{"@type":"ListItem","position":2,"name":"Python train_test_split Guide"}]},{"@type":"WebSite","@id":"https:\/\/www.silicloud.com\/blog\/#website","url":"https:\/\/www.silicloud.com\/blog\/","name":"Silicon Cloud Blog","description":"","publisher":{"@id":"https:\/\/www.silicloud.com\/blog\/#organization"},"inLanguage":"en-US"},{"@type":"Organization","@id":"https:\/\/www.silicloud.com\/blog\/#organization","name":"Silicon Cloud Blog","url":"https:\/\/www.silicloud.com\/blog\/","logo":{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/","url":"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png","contentUrl":"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png","width":1024,"height":1024,"caption":"Silicon Cloud Blog"},"image":{"@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/"},"sameAs":["https:\/\/www.facebook.com\/SiliCloudGlobal\/","https:\/\/twitter.com\/SiliCloudGlobal"]},{"@type":"Person","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/5579144e23c225c8188167f3e3f888dd","name":"Isabella Edwards","image":{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/image\/","url":"https:\/\/secure.gravatar.com\/avatar\/d4d4dec47f553ac7961d9fa4cc9bdcdcf5b7ce5106594330b6d25c5694fdbaec?s=96&d=mm&r=g","contentUrl":"https:\/\/secure.gravatar.com\/avatar\/d4d4dec47f553ac7961d9fa4cc9bdcdcf5b7ce5106594330b6d25c5694fdbaec?s=96&d=mm&r=g","caption":"Isabella Edwards"},"url":"https:\/\/www.silicloud.com\/blog\/author\/isabellaedwards\/"}]}},"_links":{"self":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/10647","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/users\/13"}],"replies":[{"embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/comments?post=10647"}],"version-history":[{"count":2,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/10647\/revisions"}],"predecessor-version":[{"id":154413,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/10647\/revisions\/154413"}],"wp:attachment":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/media?parent=10647"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/categories?post=10647"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/tags?post=10647"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}