{"id":53261,"date":"2025-09-18T15:18:56","date_gmt":"2025-09-18T15:18:56","guid":{"rendered":"https:\/\/www.devopsschool.com\/blog\/?p=53261"},"modified":"2025-09-18T15:18:56","modified_gmt":"2025-09-18T15:18:56","slug":"databricks-hello-world-ai-code","status":"publish","type":"post","link":"https:\/\/www.devopsschool.com\/blog\/databricks-hello-world-ai-code\/","title":{"rendered":"Databricks: Hello world AI Code"},"content":{"rendered":"<pre class=\"wp-block-code\" aria-describedby=\"shcb-language-1\" data-shcb-language-name=\"PHP\" data-shcb-language-slug=\"php\"><span><code class=\"hljs language-php\"><span class=\"hljs-comment\"># Hello World ML on Databricks<\/span>\n\nimport mlflow\nimport mlflow.sklearn\nfrom mlflow.models import infer_signature\nfrom sklearn.datasets import load_iris\nfrom sklearn.linear_model import LogisticRegression\n\n<span class=\"hljs-comment\"># 1. Load toy data<\/span>\nX, y = load_iris(return_X_y=<span class=\"hljs-keyword\">True<\/span>, as_frame=<span class=\"hljs-keyword\">True<\/span>)\n\n<span class=\"hljs-comment\"># 2. Train a tiny model<\/span>\nmodel = LogisticRegression(max_iter=<span class=\"hljs-number\">100<\/span>).fit(X, y)\npreds = model.predict(X)\n\n<span class=\"hljs-comment\"># 3. Set experiment (will auto-create if not present)<\/span>\nmlflow.set_experiment(<span class=\"hljs-string\">\"\/Shared\/hello_world_exp\"<\/span>)\n\n<span class=\"hljs-comment\"># 4. Log model with signature + input_example<\/span>\nwith mlflow.start_run():\n    sig = infer_signature(X, preds)\n    mlflow.sklearn.log_model(\n        sk_model=model,\n        artifact_path=<span class=\"hljs-string\">\"model\"<\/span>,\n        signature=sig,\n        input_example=X.head(<span class=\"hljs-number\">3<\/span>),\n        registered_model_name=<span class=\"hljs-string\">\"hello_world_model\"<\/span>\n    )\n<\/code><\/span><small class=\"shcb-language\" id=\"shcb-language-1\"><span class=\"shcb-language__label\">Code language:<\/span> <span class=\"shcb-language__name\">PHP<\/span> <span class=\"shcb-language__paren\">(<\/span><span class=\"shcb-language__slug\">php<\/span><span class=\"shcb-language__paren\">)<\/span><\/small><\/pre>","protected":false},"excerpt":{"rendered":"","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"_joinchat":[],"footnotes":""},"categories":[2],"tags":[],"class_list":["post-53261","post","type-post","status-publish","format-standard","hentry","category-uncategorised"],"_links":{"self":[{"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/posts\/53261","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/comments?post=53261"}],"version-history":[{"count":1,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/posts\/53261\/revisions"}],"predecessor-version":[{"id":53262,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/posts\/53261\/revisions\/53262"}],"wp:attachment":[{"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/media?parent=53261"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/categories?post=53261"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.devopsschool.com\/blog\/wp-json\/wp\/v2\/tags?post=53261"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}