Commit ea7db889 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent 6b0cc23a
Loading
Loading
Loading
Loading
+0 −4
Original line number Diff line number Diff line
"""
Script that trains DAG models on delaney dataset.
"""
from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals

import numpy as np
np.random.seed(123)
import tensorflow as tf
+0 −7
Original line number Diff line number Diff line
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Sat Jul 29 23:49:02 2017

@author: zqwu
"""

from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals

import numpy as np
np.random.seed(123)
import tensorflow as tf
+0 −4
Original line number Diff line number Diff line
@@ -5,10 +5,6 @@ Created on Sun Jul 23 15:04:19 2017

@author: zqwu
"""

from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals
from sklearn.kernel_ridge import KernelRidge

import numpy as np
+1 −5
Original line number Diff line number Diff line
"""
Script that trains textCNN models on delaney dataset.
"""
from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals

import numpy as np
np.random.seed(123)
import tensorflow as tf
@@ -34,7 +30,7 @@ model = dc.models.TextCNNModel(
    use_queue=False)

# Fit trained model
model.fit(train_dataset, nb_epoch=100)
model.fit(train_dataset, nb_epoch=10)

print("Evaluating model")
train_scores = model.evaluate(train_dataset, [metric], transformers)
+0 −4
Original line number Diff line number Diff line
"""
Script that trains weave models on delaney dataset.
"""
from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals

import numpy as np
np.random.seed(123)
import tensorflow as tf
Loading