mirror of
https://github.com/kashifulhaque/smoltorch.git
synced 2025-12-06 07:02:51 +00:00
camera ready version
This commit is contained in:
@@ -2,8 +2,8 @@ import numpy as np
|
||||
from sklearn.datasets import load_breast_cancer
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from nanotorch.tensor import Tensor
|
||||
from nanotorch.nn import MLP, SGD, binary_cross_entropy
|
||||
from smoltorch.tensor import Tensor
|
||||
from smoltorch.nn import MLP, SGD, binary_cross_entropy
|
||||
|
||||
# Load breast cancer dataset (binary classification)
|
||||
print("Loading breast cancer dataset...")
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
from sklearn.datasets import make_regression
|
||||
from sklearn.model_selection import train_test_split
|
||||
from nanotorch.tensor import Tensor
|
||||
from nanotorch.nn import MLP, SGD
|
||||
from smoltorch.tensor import Tensor
|
||||
from smoltorch.nn import MLP, SGD
|
||||
|
||||
# Generate synthetic regression data
|
||||
print("Generating data...")
|
||||
|
||||
Reference in New Issue
Block a user