import os class MyPath(object): @staticmethod def db_root_dir(database='wjd'): db_names = ['Ag', 'Al', 'Au', 'Cu', 'Ta', 'Mo', 'V', 'W'] assert (database in db_names) if database == 'Ag': return 'D:/wjd/Ag/' elif database == 'Al': return 'D:/wjd/Al/' elif database == 'Au': return 'D:/wjd/Au/' elif database == 'Cu': return 'D:/wjd/Cu/' elif database == 'Ta': return 'D:/wjd/Ta/' elif database == 'Mo': return 'D:/wjd/Mo/' elif database == 'V': return 'D:/wjd/V/' elif database == 'W': return 'D:/wjd/W/' else: raise NotImplementedError,STL_ROOT = 'D:/wjd/',应该怎么改过来啊
时间: 2024-02-26 20:54:37 浏览: 101
文件::meat_on_bone:用于Go测试的小型临时文件实用程序
您可以将 `MyPath` 类中的 `db_root_dir` 方法改为以下内容:
```
@staticmethod
def db_root_dir(database='stl-10'):
if database == 'stl-10':
return STL_ROOT
elif database in ['Ag', 'Al', 'Au', 'Cu', 'Ta', 'Mo', 'V', 'W']:
return os.path.join(STL_ROOT, database)
else:
raise NotImplementedError
```
这样,如果您调用 `MyPath.db_root_dir()` 方法并将 `database` 参数设置为 `'stl-10'`,则会返回 `STL_ROOT` 变量所指定的路径;如果您将 `database` 参数设置为其他字符串,则会返回对应数据库的路径,该路径由 `STL_ROOT` 和数据库名称组成。同时,请注意将 `STL_ROOT` 变量设置为 `'D:/wjd/'`,这是您数据集的根目录路径。
阅读全文