使用机器学习搭建深度学习实验时,通常需要在界面右侧设置读取目录、代码文件等参数。这些参数通过“—XXX”(XXX代表字符串)的形式传入,tf.flags提供了这个功能。
列出oss桶下所有的csv文件:
import tensorflow as tf
import os
FLAGS = tf.flags.FLAGS
tf.flags.DEFINE_string('buckets', 'oss://myhaspl-ai.oss-cn-beijing-internal.aliyuncs.com/', '')
tf.flags.DEFINE_string('batch_size', '15', 'batch大小')
files = tf.gfile.Glob(os.path.join(FLAGS.buckets,'*.csv')) # 如我想列出buckets下所有csv文件路径
with tf.Session() as sess:
print files
阿里云机器学习PAI读取OSS文件
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Sat Sep 15 10:54:53 2018
@author: myhaspl
@email:myhaspl@myhaspl.com
阿里云读取文件
csv格式:怀孕次数、葡萄糖、血压、皮肤厚度,胰岛素,bmi,糖尿病血统函数,年龄,结果
"""
import tensorflow as tf
import os
sampleCount=200
testCount=10
g=tf.Graph()
with g.as_default():
def inputFromFile(fileName,skipLines=