Spaces:

templates
/

tfjs

Running

osanseviero commited on Mar 10, 2022

Commit

9c0f9ca

1 Parent(s): 9d43667

Update index.html

Files changed (1) hide show

index.html CHANGED Viewed

@@ -1,24 +1,27 @@
-<!DOCTYPE html>
 <html>
-	<head>
-		<meta charset="utf-8" />
-		<meta name="viewport" content="width=device-width" />
-		<title>My static Space</title>
-		<link rel="stylesheet" href="style.css" />
-	</head>
-	<body>
-		<div class="card">
-			<h1>Welcome to your static Space!</h1>
-			<p>
-				You can modify this app directly by editing <i>index.html</i> in the
-				Files and versions tab.
-			</p>
-			<p>
-				Also don't forget to check the
-				<a href="https://huggingface.co/docs/hub/spaces" target="_blank"
-					>Spaces documentation</a
-				>.
-			</p>
-		</div>
-	</body>
-</html>

 <html>
+ <head>
+   <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.11.2"></script>
+ </head>
+ <body>
+   <div id="output_field"></div>
+ </body>
+ <script>
+ async function learnLinear(){
+  const model = tf.sequential();
+  model.add(tf.layers.dense({units: 1, inputShape: [1]}));
+  model.compile({
+   loss: 'meanSquaredError',
+   optimizer: 'sgd'
+  });
+  const xs = tf.tensor2d([-1, 0, 1, 2, 3, 4], [6, 1]);
+  const ys = tf.tensor2d([-3, -1, 1, 3, 5, 7], [6, 1]);
+  await model.fit(xs, ys, {epochs: 500});
+  document.getElementById('output_field').innerText =
+   model.predict(tf.tensor2d([10], [1, 1]));
+ }
+ learnLinear();
+ </script>
+<html>