Dodanie obsługi parametru epochs do lab 06 zad 1
This commit is contained in:
parent
6462a5fce5
commit
d04500d1d2
@ -1,5 +1,5 @@
|
||||
## Klasyfikacja jakości diamentu
|
||||
import os
|
||||
import argparse
|
||||
import pandas as pd
|
||||
import numpy as np
|
||||
import pickle
|
||||
@ -10,8 +10,11 @@ from tensorflow.keras.callbacks import History
|
||||
from sklearn.preprocessing import LabelEncoder, StandardScaler, OneHotEncoder
|
||||
from tensorflow.keras.utils import to_categorical
|
||||
|
||||
#Wyświetlenie zbioru danych
|
||||
epochs = int(os.environ.get('EPOCHS', 10))
|
||||
# Wczytanie parametru epochs
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--epochs", type=int, default=10, help="Number of epochs")
|
||||
args = parser.parse_args()
|
||||
epochs = args.epochs
|
||||
|
||||
# Wczytanie danych
|
||||
data_train = pd.read_csv('dane/diamonds_train.csv')
|
||||
|
Loading…
Reference in New Issue
Block a user