Numpy-Neuron / nn /train.py
Jensen-holm's picture
handling errors well with the neural netork api
8c348c5
raw
history blame
329 Bytes
from sklearn.model_selection import train_test_split
from nn.nn import NN
import pandas as pd
import numpy as np
def train(nn: NN) -> dict:
X_train, X_test, y_train, y_test = train_test_split(
nn.X,
nn.y,
test_size=nn.test_size,
random_state=88,
)
return {"status": "you made it!"}