diff --git a/demo/fit_a_line/train.py b/demo/fit_a_line/train.py new file mode 100644 index 00000000..8449e456 --- /dev/null +++ b/demo/fit_a_line/train.py @@ -0,0 +1,49 @@ +import paddle.v2 as paddle +import pcloud.dataset.uci_housing as uci_housing + +def main(): + # init + paddle.init() + + # network config + x = paddle.layer.data(name='x', type=paddle.data_type.dense_vector(13)) + y_predict = paddle.layer.fc(input=x, size=1, act=paddle.activation.Linear()) + y = paddle.layer.data(name='y', type=paddle.data_type.dense_vector(1)) + cost = paddle.layer.mse_cost(input=y_predict, label=y) + + # create parameters + parameters = paddle.parameters.create(cost) + + # create optimizer + optimizer = paddle.optimizer.Momentum(momentum=0) + + trainer = paddle.trainer.SGD( + cost=cost, parameters=parameters, update_equation=optimizer) + + feeding = {'x': 0, 'y': 1} + + # event_handler to print training and testing info + def event_handler(event): + if isinstance(event, paddle.event.EndIteration): + if event.batch_id % 100 == 0: + print "Pass %d, Batch %d, Cost %f" % ( + event.pass_id, event.batch_id, event.cost) + + if isinstance(event, paddle.event.EndPass): + result = trainer.test( + reader=paddle.batch(uci_housing.test(), batch_size=2), + feeding=feeding) + print "Test %d, Cost %f" % (event.pass_id, result.cost) + + # training + trainer.train( + reader=paddle.batch( + paddle.reader.shuffle(uci_housing.train(), buf_size=500), + batch_size=2), + feeding=feeding, + event_handler=event_handler, + num_passes=30) + + +if __name__ == '__main__': + main() diff --git a/docker/.gitignore b/docker/.gitignore new file mode 100644 index 00000000..f3c1a937 --- /dev/null +++ b/docker/.gitignore @@ -0,0 +1,3 @@ +Dockerfile +*.pyc +pcloud.egg-info diff --git a/docker/cloud-job/README.md b/docker/README.md similarity index 100% rename from docker/cloud-job/README.md rename to docker/README.md diff --git a/docker/cloud-job/build_docker.sh b/docker/build_docker.sh similarity index 64% rename from docker/cloud-job/build_docker.sh rename to docker/build_docker.sh index 00f2f1cb..ba825c29 100755 --- a/docker/cloud-job/build_docker.sh +++ b/docker/build_docker.sh @@ -18,12 +18,20 @@ fi echo "base_image": $base_image echo "pcloudjob_image": $pcloudjob_image +#Build Python Package +docker run --rm -it -v $PWD:/cloud $base_image \ + bash -c "cd /cloud/python && python setup.py bdist_wheel" + #Build Docker Image cat > Dockerfile <