16 lines
362 B
Python
16 lines
362 B
Python
|
from sklearn.model_selection import train_test_split
|
||
|
|
|
||
|
|
from dataset import load_dataset
|
||
|
|
from lasso_test import lasso_test
|
||
|
|
|
||
|
|
|
||
|
|
# Load dataset
|
||
|
|
X, Y, name = load_dataset()
|
||
|
|
|
||
|
|
# Split dataset
|
||
|
|
X_train, X_test, y_train, y_test = train_test_split(
|
||
|
|
X, Y, test_size=0.05, random_state=42)
|
||
|
|
|
||
|
|
for i in range(6):
|
||
|
|
lasso_test(X_train, X_test, y_train, y_test, i/pow(10, i))
|