10 lines
335 B
Python
10 lines
335 B
Python
|
import pandas as pd
|
||
|
|
import data.maker
|
||
|
|
|
||
|
|
df = pd.read_csv('sample.csv')
|
||
|
|
column = 'gender'
|
||
|
|
id = 'id'
|
||
|
|
context = 'demo'
|
||
|
|
store = {"type":"mongo.MongoWriter","args":{"host":"localhost:27017","dbname":"GAN"}}
|
||
|
|
max_epochs = 11
|
||
|
|
data.maker.train(store=store,max_epochs=max_epochs,context=context,data=df,column=column,id=id,logs='foo')
|